Update app.py

app.py

@@ -1,38 +1,67 @@
-import gradio as gr
-import openai
+import os
 import requests
+import gradio as gr
+from openai import OpenAI
 
-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"
+# 🔑 API key-үүдийг системээс авах
+openai_api_key = os.getenv("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")
+gemini_api_key = "AIzaSyCOREn4MTSkCPRK_0BF-CY6qpFEmMda64E"
 
-def fusion_response(message):
-    # ChatGPT хариулт (new API)
+# ✅ OpenAI client
+client = OpenAI(api_key=openai_api_key)
+
+# 🧠 ChatGPT function
+def chatgpt_response(prompt):
     try:
-        response = openai.chat.completions.create(
-            model="gpt-4",
-            messages=[{"role":"user","content": message}],
-            api_key=OPENAI_API_KEY
+        completion = client.chat.completions.create(
+            model="gpt-3.5-turbo",
+            messages=[{"role": "user", "content": prompt}]
         )
-        chatgpt_resp = response.choices[0].message.content
+        return completion.choices[0].message.content
     except Exception as e:
-        chatgpt_resp = f"ChatGPT error: {e}"
+        return f"ChatGPT error: {e}"
 
-    # Gemini 2 хариулт
+# 🤖 Gemini function
+def gemini_response(prompt):
     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_text = gemini_resp.get("parts", [{}])[0].get("text", "No response")
+        url = f"https://generativelanguage.googleapis.com/v1beta/models/gemini-2.0-flash:generateContent?key={gemini_api_key}"
+        headers = {"Content-Type": "application/json"}
+        data = {
+            "contents": [{"parts": [{"text": prompt}]}]
+        }
+        res = requests.post(url, headers=headers, json=data)
+        res_json = res.json()
+        return res_json["candidates"][0]["content"]["parts"][0]["text"]
     except Exception as e:
-        gemini_text = f"Gemini error: {e}"
+        return f"Gemini error: {e}"
+
+# ⚡ Fusion Logic (хоёулангийн хариултыг нэгтгэнэ)
+def fusion_ai(prompt):
+    gpt_reply = chatgpt_response(prompt)
+    gemini_reply = gemini_response(prompt)
+
+    if "error" in gpt_reply.lower() and "error" in gemini_reply.lower():
+        return "⚠️ Аль аль AI хариу өгөөгүй байна. API key эсвэл интернетээ шалгаарай."
+
+    return f"""
+🧩 *Fusion Summary*
+ChatGPT 🧠: {gpt_reply}
+
+Gemini 🌐: {gemini_reply}
 
-    fusion_text = f"ChatGPT: {chatgpt_resp}\nGemini: {gemini_text}"
-    return fusion_text
+✨ *ZeppFusion Combined Insight:*
+{gpt_reply if len(gpt_reply) > len(gemini_reply) else gemini_reply}
+"""
 
-with gr.Blocks() as demo:
-    txt = gr.Textbox(label="Message", placeholder="Энд бичнэ үү...")
-    out = gr.Textbox(label="Fusion Response")
-    txt.submit(fusion_response, txt, out)
+# 💬 Gradio Interface
+app = gr.Interface(
+    fn=fusion_ai,
+    inputs=gr.Textbox(label="🗨️ Та асуултаа бичнэ үү"),
+    outputs=gr.Markdown(label="💡 ZeppFusion AI-ийн хариулт"),
+    title="🚀 ZeppFusion AI",
+    description="ChatGPT + Gemini хослуулсан Fusion AI chatbot",
+    theme="soft",
+)
 
-demo.launch()
+if _name_ == "_main_":
+    app.launch()