Update app.py

app.py

@@ -5,6 +5,7 @@ import os
 import uuid
 from datetime import datetime
 import time
+import re
 
 # Load environment variables
 load_dotenv()
@@ -400,7 +401,7 @@ if prompt:
     if len(messages) == 0:
         # Check if the first prompt is just a greeting
         greetings = ["hello", "hi", "hey", "greetings", "good morning", "good afternoon", "good evening", "howdy"]
-        is_greeting = any(greeting in prompt.lower() for greeting in greetings)
+        is_greeting = any(greeting.lower() in prompt.lower() for greeting in greetings) and len(prompt.split()) < 5
         
         # Keep "New chat" if it's a greeting, otherwise change immediately after first response
         if not is_greeting:
@@ -408,7 +409,7 @@ if prompt:
             st.session_state.conversations[current_id]["should_update_title"] = True
     
     # Update title after 2nd or 3rd user message if still has default title
-    elif len(messages) >= 3 and st.session_state.conversations[current_id]["title"].startswith("New chat"):
+    elif 1 <= len(messages) <= 5 and st.session_state.conversations[current_id]["title"].startswith("New chat"):
         # Create a short 3-4 word summary
         words = prompt.split()
         if len(words) > 4:
@@ -449,7 +450,13 @@ if prompt:
                 
                 # Final display without cursor
                 if full_response:
-                    message_placeholder.markdown(full_response)
+                    # Clean up any excessive dash lines or formatting issues
+            cleaned_response = full_response
+            # Replace multiple consecutive dash lines with a single line
+            cleaned_response = re.sub(r'━{5,}', '━━━━━', cleaned_response)
+            # Remove excessive newlines
+            cleaned_response = re.sub(r'\n{3,}', '\n\n', cleaned_response)
+            message_placeholder.markdown(cleaned_response)
                 else:
                     message_placeholder.markdown("No response received from the model.")
             
@@ -458,21 +465,29 @@ if prompt:
             
             # Check if we should update the title after first response
             if len(messages) == 2 and st.session_state.conversations[current_id].get("should_update_title", False):
-                # Extract a short title from the conversation context
-                combined_text = prompt + " " + full_response
-                words = combined_text.split()
-                title_words = words[:4] if len(words) > 4 else words
-                new_title = " ".join(title_words)
+                # Extract a short title from the first user prompt
+                user_prompt = messages[0]["content"]
+                words = user_prompt.split()
+                if len(words) >= 4:
+                    # Use first 3-4 meaningful words for title
+                    title_words = words[:4] 
+                    new_title = " ".join(title_words)
+                else:
+                    new_title = user_prompt
+                    
                 st.session_state.conversations[current_id]["title"] = new_title[:30]
                 st.session_state.conversations[current_id]["should_update_title"] = False
+                
+            # Force a refresh of the UI after each message to ensure title updates
+            st.experimental_rerun()
             
         except Exception as e:
             error_msg = f"Error: {str(e)}"
             message_placeholder.markdown(error_msg)
             messages.append({"role": "assistant", "content": error_msg})
     
-    # Force rerun to update the UI
-    st.rerun()
+    # We will force rerun after message processing completes in the assistant block above
+    # This allows us to update the UI after title changes
 
 # JavaScript for Auto-Scroll - Improved version
 scroll_script = """