Update app.py

app.py

@@ -144,8 +144,7 @@ def generate_response(user_input, top_p, temperature, chat_counter, chatbot, his
     messages = [
         {"role": "system", "content": "You are a helpful AI assistant. Provide clear, friendly, and informative responses to user questions. Analyze any provided images, videos, or web content to inform your response."}
     ]
-    for msg in history:
-        messages.append(msg)
+    messages += history  # OpenAI-style history
 
     # Prepare multimodal content
     multimodal_content = []
@@ -176,34 +175,37 @@ def generate_response(user_input, top_p, temperature, chat_counter, chatbot, his
         if content_type == 'webpage':
             web_content = extract_webpage_content(url)
             multimodal_content.append({"type": "text", "text": f"[Content from URL ({url})]:\n{web_content}"})
-        elif content_type == 'image':
-             multimodal_content.append({"type": "image_url", "image_url": {"url": url}})
+        elif content_type.startswith("image"):
+            multimodal_content.append({"type": "image_url", "image_url": {"url": url}})
         else:
             multimodal_content.append({"type": "text", "text": f"[URL content of type '{content_type}' is not supported for direct analysis. Please describe the content if you can access it.]"})
 
+    # Add the new user message to the conversation
     messages.append({"role": "user", "content": multimodal_content})
 
-    # Generate response
+    # Generate response from OpenAI
     response_text = generate_chatgpt_response(messages, temperature)
 
-    # Translate response if necessary
+    # Translate back if needed
     if lang == "bg":
         response_text = en_to_bg(response_text)[0]["translation_text"]
 
-    # Update history
+    # Update OpenAI-style history
     history.append({"role": "user", "content": user_input})
     history.append({"role": "assistant", "content": response_text})
 
+    # ✅ Convert OpenAI-style history to Chatbot-style list of [user, bot] pairs
     chatbot_messages = []
     for i in range(0, len(history), 2):
-        if i + 1 < len(history):
+        try:
             user_msg = history[i]["content"]
             bot_msg = history[i + 1]["content"]
             chatbot_messages.append([user_msg, bot_msg])
-
+        except IndexError:
+            pass  # incomplete pair; skip
 
     chat_counter += 1
-    return history, history, chat_counter, "✅ Success", gr.update(value="", interactive=True), gr.update(interactive=True)
+    return chatbot_messages, history, chat_counter, "✅ Success", gr.update(value="", interactive=True), gr.update(interactive=True)
 
 # --- Gradio UI ---
 
@@ -214,7 +216,6 @@ def generate_response(user_input, top_p, temperature, chat_counter, chatbot, his
 #     demo.queue(max_size=10).launch(server_name="0.0.0.0", server_port=7860)
 
 
-
 def reset_textbox():
     return gr.update(value="", interactive=True)