Update app.py

app.py

@@ -9,34 +9,34 @@ def create_conversation_prompt(name1: str, name2: str, persona_style: str):
     """
     Create a prompt that instructs the model to produce exactly 15 messages
     of conversation, alternating between name1 and name2, starting with name1.
+
+    We will be very explicit and not allow any formatting except the required lines.
     """
     prompt_template_str = f"""
-    You are to simulate a conversation of exactly 15 messages total between two people: {name1} and {name2}.
-    The conversation should reflect the style: {persona_style}.
-    {name1} speaks first (message 1), then {name2} (message 2), then {name1} (message 3), and so forth, 
-    alternating until all 15 messages are complete (the 15th message is by {name1}).
-
-    Rules:
-    - Output only the 15 messages of the conversation, nothing else.
-    - Each message must be in the format:
-      {name1}: <one or two short sentences>
-      {name2}: <one or two short sentences>
-    - Do not add any headings, such as '# Conversation' or '---'.
-    - Do not repeat the conversation.
-    - Do not produce anything after the 15th message.
+    You are simulating a conversation of exactly 15 messages between two people: {name1} and {name2}.
+    {name1} speaks first (message 1), then {name2} (message 2), then {name1} (message 3), and so forth,
+    alternating until all 15 messages are complete. The 15th message is by {name1}.
+
+    Requirements:
+    - Output exactly 15 lines, no more, no less.
+    - Each line must be a single message in the format:
+      {name1}: <message> or {name2}: <message>
+    - Do not add any headings, numbers, sample outputs, or explanations.
+    - Do not mention code, programming, or instructions.
+    - Each message should be 1-2 short sentences, friendly, natural, reflecting the style: {persona_style}.
     - Use everyday language, can ask questions, show opinions.
-    - Use emojis sparingly, if at all (no more than 1-2 in the entire conversation).
-    - No repetition of the same message. Each message is unique and responds logically to the previous one.
-    - The conversation ends exactly after the 15th message by {name1} with no extra lines or content.
+    - Use emojis sparingly if it fits the style (no more than 1-2 total).
+    - No repeated lines, each message should logically follow from the previous one.
+    - Do not produce anything after the 15th message. No extra lines or text.
 
-    Now, produce all 15 messages strictly following the above instructions:
+    Produce all 15 messages now:
     """
     return ChatPromptTemplate.from_template(prompt_template_str)
 
 def create_summary_prompt(name1: str, name2: str, conversation: str):
-    """Create a prompt specifically for generating a title and summary of the conversation."""
+    """Prompt for generating a title and summary."""
     summary_prompt_str = f"""
-    The following is a completed conversation between {name1} and {name2}:
+    Below is a completed 15-message conversation between {name1} and {name2}:
 
     {conversation}
 
@@ -44,7 +44,9 @@ def create_summary_prompt(name1: str, name2: str, conversation: str):
     Title: <A short descriptive title of the conversation>
     Summary: <A few short sentences highlighting the main points, tone, and conclusion>
 
-    Do not continue the conversation, do not repeat it, just provide a title and summary.
+    Do not continue the conversation, do not repeat it, and do not add extra formatting beyond the two lines:
+    - One line starting with "Title:"
+    - One line starting with "Summary:"
     """
     return ChatPromptTemplate.from_template(summary_prompt_str)
 
@@ -53,11 +55,9 @@ def main():
 
     model_names = [
         "meta-llama/Llama-3.3-70B-Instruct",
-        "mistralai/Mistral-7B-v0.1",
+        "meta-llama/Llama-3.1-405B-Instruct",
         "lmsys/vicuna-13b-v1.5",
-        "tiiuae/falcon-180B",
-        "EleutherAI/gpt-neox-20b",
-        "dice-research/lola_v1"
+        "mistral-7b"
     ]
     selected_model = st.selectbox("Select a model:", model_names)
 
@@ -96,15 +96,14 @@ def main():
 
             try:
                 # Generate all 15 messages in one go
-                conversation = conversation_chain.run(chat_history="", input="Produce the full conversation now.")
-                conversation = conversation.strip()
+                conversation = conversation_chain.run(chat_history="", input="").strip()
 
                 st.subheader("Final Conversation:")
                 st.text(conversation)
                 print("Conversation Generation Complete.\n")
                 print("Full Conversation:\n", conversation)
 
-                # Now summarize
+                # Summarize the conversation
                 summary_prompt = create_summary_prompt(name1, name2, conversation)
                 summary_chain = LLMChain(llm=llm, prompt=summary_prompt)
 
@@ -122,4 +121,3 @@ def main():
 
 if __name__ == "__main__":
     main()
-