Update app.py

app.py

@@ -2,30 +2,42 @@ import gradio as gr
 import openai
 import requests
 
+# API түлхүүрүүд
 OPENAI_API_KEY = "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"
 GEMINI_API_KEY = "AIzaSyCOREn4MTSkCPRK_0BF-CY6qpFEmMda64E"
+GEMINI_ENDPOINT = "https://generativelanguage.googleapis.com/v1beta/models/gemini-2.0-flash:generateContent"
 
-def fusion_response(message, mode="Fusion"):
-    chatgpt_resp = openai.ChatCompletion.create(
-        model="gpt-4",
-        messages=[{"role":"user","content":message}]
-    )["choices"][0]["message"]["content"]
+def fusion_response(message):
+    # ChatGPT хариулт
+    try:
+        chatgpt_resp = openai.ChatCompletion.create(
+            model="gpt-4",
+            messages=[{"role":"user","content": message}],
+            api_key=OPENAI_API_KEY
+        )["choices"][0]["message"]["content"]
+    except Exception as e:
+        chatgpt_resp = f"ChatGPT error: {e}"
 
-    gemini_resp = requests.post(
-        "https://gemini-api-url",
-        headers={"Authorization": f"Bearer {GEMINI_API_KEY}"},
-        json={"prompt": message}
-    ).json()["reply"]
+    # Gemini 2 хариулт
+    try:
+        payload = {
+            "contents": [{"parts":[{"text": message}]}]
+        }
+        headers = {"Content-Type": "application/json", "X-goog-api-key": GEMINI_API_KEY}
+        gemini_resp = requests.post(GEMINI_ENDPOINT, headers=headers, json=payload).json()
+        # Gemini response-г текстээр авах
+        gemini_text = gemini_resp.get("candidates", [{}])[0].get("content", "No response")
+    except Exception as e:
+        gemini_text = f"Gemini error: {e}"
 
-    if mode=="ChatGPT": return chatgpt_resp
-    if mode=="Gemini": return gemini_resp
-    # Fusion mode
-    return f"ChatGPT: {chatgpt_resp}\nGemini: {gemini_resp}"
+    # Хоёрыг нэгтгэн буцаах
+    fusion_text = f"ChatGPT: {chatgpt_resp}\nGemini: {gemini_text}"
+    return fusion_text
 
+# Gradio интерфейс
 with gr.Blocks() as demo:
-    txt = gr.Textbox(label="Message")
-    out = gr.Textbox(label="Response")
-    mode = gr.Dropdown(["ChatGPT","Gemini","Fusion"], value="Fusion", label="Mode")
-    txt.submit(fusion_response, [txt, mode], out)
+    txt = gr.Textbox(label="Message", placeholder="Энд бичнэ үү...")
+    out = gr.Textbox(label="Fusion Response")
+    txt.submit(fusion_response, txt, out)
 
 demo.launch()