Update app.py

app.py

@@ -3,14 +3,20 @@ import requests
 import gradio as gr
 from openai import OpenAI
 
-# 🔑 API key-үүдийг системээс авах
-openai_api_key = os.getenv("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")
-gemini_api_key = "AIzaSyCOREn4MTSkCPRK_0BF-CY6qpFEmMda64E"
+# 🔑 API key-г environment variable-аас унших
+openai_api_key = os.getenv("OPENAI_API_KEY")
+gemini_api_key = os.getenv("GEMINI_API_KEY")
+
+# 🔍 Баталгаажуулах (лог)
+print("OPENAI_API_KEY =", openai_api_key)
+print("GEMINI_API_KEY =", gemini_api_key)
 
 # ✅ OpenAI client
+if openai_api_key is None:
+    raise ValueError("⚠️ OPENAI_API_KEY environment variable-ийг тохируулна уу!")
 client = OpenAI(api_key=openai_api_key)
 
-# 🧠 ChatGPT function
+# 🧠 ChatGPT хариу
 def chatgpt_response(prompt):
     try:
         completion = client.chat.completions.create(
@@ -19,23 +25,23 @@ def chatgpt_response(prompt):
         )
         return completion.choices[0].message.content
     except Exception as e:
-        return f"ChatGPT error: {e}"
+        return f"⚠️ ChatGPT error: {e}"
 
-# 🤖 Gemini function
+# 🤖 Gemini хариу
 def gemini_response(prompt):
+    if gemini_api_key is None:
+        return "⚠️ GEMINI_API_KEY тохируулаагүй байна!"
     try:
         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}]}]
-        }
+        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:
-        return f"Gemini error: {e}"
+        return f"⚠️ Gemini error: {e}"
 
-# ⚡ Fusion Logic (хоёулангийн хариултыг нэгтгэнэ)
+# ⚡ Fusion Logic
 def fusion_ai(prompt):
     gpt_reply = chatgpt_response(prompt)
     gemini_reply = gemini_response(prompt)
@@ -44,23 +50,32 @@ def fusion_ai(prompt):
         return "⚠️ Аль аль AI хариу өгөөгүй байна. API key эсвэл интернетээ шалгаарай."
 
     return f"""
-🧩 *Fusion Summary*
-ChatGPT 🧠: {gpt_reply}
-
-Gemini 🌐: {gemini_reply}
+<div style="background-color:#0d1117; padding:20px; border-radius:10px; color:#e6edf3; font-family:Segoe UI">
+    <h3>🧩 <span style="color:#00d9ff">Fusion Summary</span></h3>
+    <p><b>ChatGPT 🧠:</b> {gpt_reply}</p>
+    <p><b>Gemini 🌐:</b> {gemini_reply}</p>
+    <hr style="border:1px solid #222;">
+    <h4>✨ <span style="color:#9b59b6">ZeppFusion Combined Insight:</span></h4>
+    <p>{gpt_reply if len(gpt_reply) > len(gemini_reply) else gemini_reply}</p>
+</div>
+"""
 
-✨ *ZeppFusion Combined Insight:*
-{gpt_reply if len(gpt_reply) > len(gemini_reply) else gemini_reply}
+# 🎨 Gradio Interface (Dark mode + styling)
+css = """
+.gradio-container {background-color: #0d1117 !important; color: #e6edf3;}
+h1, h3, h4, label {color: #00d9ff !important;}
+textarea, input {background-color:#161b22 !important; color:#fff !important; border-radius:10px;}
+button {background-color:#00d9ff !important; color:#000 !important; font-weight:bold; border-radius:10px;}
 """
 
-# 💬 Gradio Interface
 app = gr.Interface(
     fn=fusion_ai,
-    inputs=gr.Textbox(label="🗨️ Та асуултаа бичнэ үү"),
-    outputs=gr.Markdown(label="💡 ZeppFusion AI-ийн хариулт"),
+    inputs=gr.Textbox(label="🗨️ Та асуултаа бичнэ үү", placeholder="Жишээ нь: AI гэж юу вэ?"),
+    outputs=gr.HTML(label="💡 ZeppFusion AI-ийн хариулт"),
     title="🚀 ZeppFusion AI",
-    description="ChatGPT + Gemini хослуулсан Fusion AI chatbot",
+    description="ChatGPT 🧠 + Gemini 🌐 хослуулсан Fusion AI chatbot — Where Intelligence Unites.",
     theme="soft",
+    css=css,
 )
 
 if _name_ == "_main_":