updated the UI to have title

app.py

@@ -7,6 +7,7 @@ BACKEND_URL = "https://0learn-fastchat.hf.space"  # Update this with your actual
 class ConversationState:
     def __init__(self):
         self.conversation_id = None
+        self.title = "New Conversation"
 
 state = ConversationState()
 
@@ -22,39 +23,58 @@ def chat_with_groq(message, history):
     response_data = response.json()
     
     state.conversation_id = response_data["conversation_id"]
+    
+    # Update title if it's a new conversation
+    if len(history) == 0:
+        state.title = message[:30] + "..." if len(message) > 30 else message
+    
     return response_data["response"]
 
 def load_conversation(conversation_id):
     if not conversation_id:
-        return []
+        return [], state.title
     response = requests.get(f"{BACKEND_URL}/conversations/{conversation_id}")
     if response.status_code == 200:
         conversation_data = response.json()
         state.conversation_id = conversation_id
         messages = conversation_data["messages"]
-        return [(msg["content"], messages[i+1]["content"]) for i, msg in enumerate(messages[::2])]
+        state.title = messages[0]["content"][:30] + "..." if len(messages[0]["content"]) > 30 else messages[0]["content"]
+        return [(msg["content"], messages[i+1]["content"]) for i, msg in enumerate(messages[::2])], state.title
+    else:
+        return [], "Conversation not found"
+
+def start_new_conversation():
+    state.conversation_id = None
+    state.title = "New Conversation"
+    return [], state.title
+
+def continue_conversation():
+    if state.conversation_id:
+        return load_conversation(state.conversation_id)
     else:
-        return []
+        return [], "No previous conversation"
 
 with gr.Blocks() as demo:
-    gr.Markdown("# Groq Chatbot with Conversation Persistence")
+    gr.Markdown("# Groq Chatbot with Enhanced Conversation Management")
     
-    conversation_id_input = gr.Textbox(label="Conversation ID (optional)")
-    load_button = gr.Button("Load Conversation")
+    with gr.Row():
+        new_conv_btn = gr.Button("New Conversation")
+        continue_conv_btn = gr.Button("Continue Previous Conversation")
+    
+    title_display = gr.Markdown(f"## Current Conversation: {state.title}")
     
     chatbot = gr.Chatbot()
-    msg = gr.Textbox()
-    clear = gr.Button("Clear")
+    msg = gr.Textbox(label="Type your message here")
+    clear = gr.Button("Clear Chat")
 
     def respond(message, chat_history):
         bot_message = chat_with_groq(message, chat_history)
         chat_history.append((message, bot_message))
-        return "", chat_history
-
-    msg.submit(respond, [msg, chatbot], [msg, chatbot])
-    clear.click(lambda: None, None, chatbot, queue=False)
-    load_button.click(load_conversation, inputs=[conversation_id_input], outputs=[chatbot])
+        return "", chat_history, state.title
 
-    gr.Markdown(f"## Current Conversation ID: {state.conversation_id}")
+    msg.submit(respond, [msg, chatbot], [msg, chatbot, title_display])
+    clear.click(lambda: ([], state.title), None, [chatbot, title_display], queue=False)
+    new_conv_btn.click(start_new_conversation, None, [chatbot, title_display])
+    continue_conv_btn.click(continue_conversation, None, [chatbot, title_display])
 
 demo.launch()