Update app.py

app.py

@@ -80,20 +80,29 @@ def chatbot_response(api_key, user_inputs, history):
     user_text = user_inputs.get("text")
     user_image = user_inputs.get("image")
     
+    # Prepare user content
+    user_content = []
+    if user_text:
+        user_content.append(user_text)
+    if user_image:
+        user_content.append(user_image)
+    
     # Append the user's message to the history
-    if user_text or user_image:
-        history.append(("User", user_text, user_image))
+    if user_content:
+        history.append((user_content, None))
     
     # Get the API response
     api_result = get_api_response(api_key, user_inputs)
     
     if "error" in api_result:
         ai_message = f"Error: {api_result['error']}"
+        bot_content = [ai_message]
     else:
         ai_message = api_result["response"]
+        bot_content = [ai_message]
     
     # Append the AI's response to the history
-    history.append(("AI", ai_message, None))
+    history[-1] = (history[-1][0], bot_content)
     
     return history, history
 
@@ -145,4 +154,4 @@ with gr.Blocks() as demo:
 
 # Launch the Gradio app
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()