Update app.py

app.py

@@ -136,7 +136,8 @@ if "current_conversation_id" not in st.session_state:
     st.session_state.current_conversation_id = new_id
     st.session_state.conversations[new_id] = {
         "title": f"New chat {datetime.now().strftime('%H:%M')}",
-        "messages": []
+        "messages": [],
+        "should_update_title": False
     }
 
 # Default Model as o1
@@ -249,7 +250,8 @@ def create_new_chat():
     st.session_state.current_conversation_id = new_id
     st.session_state.conversations[new_id] = {
         "title": f"New chat {datetime.now().strftime('%H:%M')}",
-        "messages": []
+        "messages": [],
+        "should_update_title": False
     }
 
 # Functions for system prompt editing
@@ -394,16 +396,26 @@ st.markdown('</div>', unsafe_allow_html=True)
 st.markdown('</div>', unsafe_allow_html=True)
 
 if prompt:
-    # Only update the chat title after at least 2 assistant responses
-    # This gives enough context to identify the subject of the conversation
-    if len(messages) >= 4:  # At least 2 user messages and 2 assistant responses
-        current_title = st.session_state.conversations[current_id]["title"]
-        if current_title.startswith("New chat"):
-            # Get the content of the conversation to identify the subject
-            conversation_content = " ".join([msg["content"][:100] for msg in messages[:4]])
-            # Create a brief title based on the conversation content
-            new_title = prompt[:25] + "..." if len(prompt) > 25 else prompt
-            st.session_state.conversations[current_id]["title"] = new_title
+    # Check if this is the first message
+    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)
+        
+        # Keep "New chat" if it's a greeting, otherwise change immediately after first response
+        if not is_greeting:
+            # We'll update the title after getting first response
+            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"):
+        # Create a short 3-4 word summary
+        words = prompt.split()
+        if len(words) > 4:
+            short_title = " ".join(words[:4]) + "..."
+        else:
+            short_title = prompt
+        st.session_state.conversations[current_id]["title"] = short_title[:30]
     
     # Add user message to conversation
     messages.append({"role": "user", "content": prompt})
@@ -444,6 +456,16 @@ if prompt:
             # Add the response to conversation history
             messages.append({"role": "assistant", "content": full_response})
             
+            # 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)
+                st.session_state.conversations[current_id]["title"] = new_title[:30]
+                st.session_state.conversations[current_id]["should_update_title"] = False
+            
         except Exception as e:
             error_msg = f"Error: {str(e)}"
             message_placeholder.markdown(error_msg)