Update app.py

app.py

@@ -3,22 +3,16 @@ import google.generativeai as genai
 import os
 
 # ----------------- ТОХИРГОО -----------------
-# ЧУХАЛ: API түлхүүрээ орчны хувьсагч (environment variable) болгож тохируулна уу.
-# Google AI Studio-с API түлхүүрээ авна уу: https://aistudio.google.com/app/apikey
-#
-# Терминал дээрээ дараах байдлаар тохируулж болно:
-# export GOOGLE_API_KEY="ЭНД_API_ТҮЛХҮҮРЭЭ_ОРУУЛНА_УУ"
-
+# API түлхүүрийг орчны хувьсагч болгож өгнө
 try:
     genai.configure(api_key=os.environ["GOOGLE_API_KEY"])
 except KeyError:
     print("АЛДАА: GOOGLE_API_KEY орчны хувьсагч тохируулагдаагүй байна.")
-    # Хэрэв түлхүүр байхгүй бол Gradio интерфэйс дээр алдаа харуулна
     api_key_set = False
 else:
     api_key_set = True
 
-# Аюулгүй байдлын тохиргоог зөөлрүүлэх (зарим тохиолдолд хариулт блоклогдохоос сэргийлнэ)
+# Аюулгүй байдлын тохиргоо
 safety_settings = [
     {"category": "HARM_CATEGORY_HARASSMENT", "threshold": "BLOCK_NONE"},
     {"category": "HARM_CATEGORY_HATE_SPEECH", "threshold": "BLOCK_NONE"},
@@ -26,27 +20,23 @@ safety_settings = [
     {"category": "HARM_CATEGORY_DANGEROUS_CONTENT", "threshold": "BLOCK_NONE"},
 ]
 
-# Gemini Pro загварыг сонгох
+# Gemini Pro загвар сонгох
 model = genai.GenerativeModel('gemini-pro', safety_settings=safety_settings)
 
-# Чатны түүхийг хадгалах (Gradio-ийн session state)
+# Чатны түүхийг хадгалах
 chat = model.start_chat(history=[])
 
 # ----------------- GRADIO ФУНКЦ -----------------
 def gemini_chat(message, history):
     """
-    Gradio чатны интерфэйст зориулсан функц.
-    Хэрэглэгчийн зурвасыг Gemini API руу илгээж, хариултыг нь буцаана.
+    Gradio чат интерфэйст зориулсан функц
     """
     if not api_key_set:
-        yield "АЛДАА: Google API түлхүүр тохируулагдаагүй байна. Програмыг унтрааж, API түлхүүрээ тохируулаад дахин ажиллуулна уу."
+        yield "АЛДАА: Google API түлхүүр тохируулагдаагүй байна. Програмыг унтрааж, API түлхүүрээ тохируулна уу."
         return
 
     try:
-        # Mesasge-ийг загвар луу илгээнэ
         response = chat.send_message(message, stream=True)
-
-        # Хариултыг хэсэгчлэн (stream) авах
         full_response = ""
         for chunk in response:
             full_response += chunk.text
@@ -55,36 +45,30 @@ def gemini_chat(message, history):
     except Exception as e:
         yield f"Алдаа гарлаа: {e}"
 
-
 # ----------------- GRADIO ИНТЕРФЕЙС -----------------
 with gr.Blocks(theme=gr.themes.Soft(), title="Gemini Chat") as demo:
     gr.Markdown(
         """
-    # Gemini Chat 🤖
-    Энэ бол Google-ийн Gemini Pro загварыг ашигласан чатбот юм.
-    """
+        # Gemini Chat 🤖
+        Энэ бол Google-ийн Gemini Pro загварыг ашигласан чатбот юм.
+        """
     )
-    # ChatGPT-тэй ижил дизайнтай чат интерфэйс
-    chatbot = gr.ChatInterface(
-        fn=gemini_chat,
-        chatbot=gr.Chatbot(
-            height=600,
-            show_label=False,
-            avatar_images=(None, "https://www.google.com/images/branding/googlelogo/1x/googlelogo_color_272x92dp.png"),
-        ),
-        textbox=gr.Textbox(
-            placeholder="Энд бичнэ үү...",
-            container=False,
-            scale=7,
-        ),
-        title=None,
-        submit_btn="Илгээх",
-        retry_btn=None,
-        undo_btn="Буцаах",
-        clear_btn="Цэвэрлэх",
+
+    chatbot = gr.Chatbot(height=600, show_label=False, 
+                         avatar_images=(None, "https://www.google.com/images/branding/googlelogo/1x/googlelogo_color_272x92dp.png"))
+
+    msg = gr.Textbox(
+        placeholder="Энд бичнэ үү...",
+        container=False,
+        scale=7,
     )
 
+    clear_btn = gr.Button("Цэвэрлэх")
+    
+    # Функцуудийг холбох
+    msg.submit(gemini_chat, [msg, chatbot], [msg, chatbot])
+    clear_btn.click(lambda: None, None, chatbot, queue=False)
+
 # ----------------- АЖИЛЛУУЛАХ -----------------
 if __name__ == "__main__":
-    # Gradio аппликейшнийг ажиллуулах
-    demo.launch(debug=True)
+    demo.launch(server_name="0.0.0.0", server_port=7860, debug=True)