Update app.py

app.py

@@ -12,7 +12,7 @@ state = ConversationState()
 
 def chat_with_groq(message, history):
     messages = [{"role": "user" if i % 2 == 0 else "assistant", "content": msg} 
-                for i, msg in enumerate(history + [message])]
+                for i, msg in enumerate([item for sublist in history for item in sublist] + [message])]
     
     payload = {"messages": messages}
     if state.conversation_id:
@@ -25,33 +25,36 @@ def chat_with_groq(message, history):
     return response_data["response"]
 
 def load_conversation(conversation_id):
+    if not conversation_id:
+        return []
     response = requests.get(f"{BACKEND_URL}/conversations/{conversation_id}")
     if response.status_code == 200:
         conversation_data = response.json()
         state.conversation_id = conversation_id
-        return conversation_data["messages"]
+        messages = conversation_data["messages"]
+        return [(msg["content"], messages[i+1]["content"]) for i, msg in enumerate(messages[::2])]
     else:
         return []
 
-def respond(message, chat_history):
-    bot_message = chat_with_groq(message, chat_history)
-    chat_history.append((message, bot_message))
-    return "", chat_history
-
 with gr.Blocks() as demo:
     gr.Markdown("# Groq Chatbot with Conversation Persistence")
     
-    with gr.Row():
-        conversation_id_input = gr.Textbox(label="Conversation ID (optional)")
-        load_button = gr.Button("Load Conversation")
+    conversation_id_input = gr.Textbox(label="Conversation ID (optional)")
+    load_button = gr.Button("Load Conversation")
     
     chatbot = gr.Chatbot()
     msg = gr.Textbox()
     clear = gr.Button("Clear")
 
+    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])
 
+    gr.Markdown(f"## Current Conversation ID: {state.conversation_id}")
+
 demo.launch()