Update app.py

app.py

@@ -65,27 +65,29 @@ def simulate_conversation(chain: LLMChain, name1: str, name2: str, total_message
         return None
 
 def summarize_conversation(chain: LLMChain, conversation: str, name1: str, name2: str):
+    """Use the LLM to summarize the completed conversation and provide a title."""
     st.write("**Summarizing the conversation...**")
     print("Summarizing the conversation...")
 
+    # Updated prompt to clearly request a title and summary in the desired format
     summary_prompt = f"""
     The following is a conversation between {name1} and {name2}:
     {conversation}
 
-    Provide a short descriptive title for their conversation and then summarize it in a few short sentences highlighting the main points, tone, and conclusion.
-    
-    Format your answer as:
-    Title: <your conversation title>
-    Summary: <your summary here>
+    Please provide:
+    - A short and descriptive title of the conversation on one line, starting with "Title:".
+    - A summary of the conversation on the next line, starting with "Summary:".
+    The summary should be a few short sentences highlighting the main points, tone, and conclusion.
     """
 
     try:
         response = chain.run(chat_history="", input=summary_prompt)
+        # Just return the response as-is. The model should follow the instructions given.
         return response.strip()
     except Exception as e:
         st.error(f"Error summarizing conversation: {e}")
         print(f"Error summarizing conversation: {e}")
-        return "No summary available due to error."
+        return "Title: No Title\nSummary: No summary available due to error."
 
 def main():
     st.title("LLM Conversation Simulation")