Update app.py

app.py

@@ -3,7 +3,6 @@ import google.generativeai as genai
 import os
 
 # ----------------- ТОХИРГОО -----------------
-# API түлхүүрийг орчны хувьсагч болгож өгнө
 try:
     genai.configure(api_key=os.environ["GOOGLE_API_KEY"])
 except KeyError:
@@ -29,21 +28,20 @@ chat = model.start_chat(history=[])
 # ----------------- GRADIO ФУНКЦ -----------------
 def gemini_chat(message, history):
     """
-    Gradio чат интерфэйст зориулсан функц
+    Хоёр output-тэй байх: textbox (хоосон болгох), chatbot (түүх)
     """
     if not api_key_set:
-        yield "АЛДАА: Google API түлхүүр тохируулагдаагүй байна. Програмыг унтрааж, API түлхүүрээ тохируулна уу."
-        return
+        return "", history + [{"role": "system", "content": "АЛДАА: Google API түлхүүр тохируулагдаагүй байна."}]
 
     try:
         response = chat.send_message(message, stream=True)
         full_response = ""
         for chunk in response:
             full_response += chunk.text
-            yield full_response
-
+            # Stream хийх үед tuple хэлбэрээр хоёр output буцаана
+            yield "", history + [{"role": "user", "content": message}, {"role": "assistant", "content": full_response}]
     except Exception as e:
-        yield f"Алдаа гарлаа: {e}"
+        return "", history + [{"role": "system", "content": f"Алдаа гарлаа: {e}"}]
 
 # ----------------- GRADIO ИНТЕРФЕЙС -----------------
 with gr.Blocks(theme=gr.themes.Soft(), title="Gemini Chat") as demo: