Update app.py

app.py

@@ -1,45 +1,14 @@
 import os
 import requests
 import gradio as gr
-import sqlite3
-from openai import OpenAI
 
-# 🔑 API keys
-openai_api_key = os.getenv("OPENAI_API_KEY")
+# Gemini API key
 gemini_api_key = os.getenv("GEMINI_API_KEY")
+if gemini_api_key is None:
+    raise ValueError("⚠️ GEMINI_API_KEY тохируулаагүй байна!")
 
-print("OPENAI_API_KEY =", openai_api_key)
-print("GEMINI_API_KEY =", gemini_api_key)
-
-if openai_api_key is None:
-    raise ValueError("⚠️ OPENAI_API_KEY тохируулна уу!")
-
-client = OpenAI(api_key=openai_api_key)
-
-# SQLite db setup
-conn = sqlite3.connect("chat_history.db", check_same_thread=False)
-c = conn.cursor()
-c.execute("""
-CREATE TABLE IF NOT EXISTS history (
-    id INTEGER PRIMARY KEY AUTOINCREMENT,
-    user_input TEXT,
-    fusion_output TEXT
-)
-""")
-conn.commit()
-
-# ChatGPT (gpt-5)
-def chatgpt_response(prompt):
-    try:
-        response = client.responses.create(model="gpt-5", input=prompt)
-        return response.output_text
-    except Exception as e:
-        return f"⚠️ ChatGPT error: {e}"
-
-# Gemini
+# Gemini response function
 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"}
@@ -50,47 +19,33 @@ def gemini_response(prompt):
     except Exception as e:
         return f"⚠️ Gemini error: {e}"
 
-# Fusion + summary
-def fusion_ai(prompt, chat_history=[]):
-    gpt_reply = chatgpt_response(prompt)
-    gemini_reply = gemini_response(prompt)
-
-    if "ChatGPT error" in gpt_reply.lower():
-        combined = f"⚠️ ChatGPT quota дууссан тул Gemini хариу:\n\n{gemini_reply}"
-    else:
-        # Fusion summary: урт хариултаас сонгох
-        combined = f"ChatGPT 🧠: {gpt_reply}\nGemini 🌐: {gemini_reply}"
-
-    # Update SQLite history
-    c.execute("INSERT INTO history (user_input, fusion_output) VALUES (?, ?)", (prompt, combined))
-    conn.commit()
+# Chat function with history
+def chat(prompt, history=[]):
+    reply = gemini_response(prompt)
+    history.append((prompt, reply))
+    return history, history
 
-    chat_history.append((prompt, combined))
-    return chat_history, chat_history
-
-# Gradio interface
+# Custom CSS for chat design
 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-chatbot-message.user {background-color:#0f2c54; color:#fff; border-radius:15px; padding:10px;}
+.gradio-chatbot-message.bot {background-color:#1f1f1f; color:#00d9ff; border-radius:15px; padding:10px;}
 """
 
+# Gradio interface
 with gr.Blocks(css=css, theme=gr.themes.Soft()) as app:
-    gr.Markdown("## 🚀 ZeppFusion AI — Persistent Fusion Chat")
+    gr.Markdown("## 🚀 ZeppFusion AI — Gemini Chat Only")
     chatbot = gr.Chatbot()
     msg = gr.Textbox(placeholder="Таны асуулт...")
     clear = gr.Button("🗑️ Clear History")
 
-    # Submit: add to history
-    msg.submit(fusion_ai, [msg, chatbot], [chatbot, chatbot])
+    msg.submit(chat, [msg, chatbot], [chatbot, chatbot])
     
-    # Clear button: delete SQLite history
-    def clear_history():
-        c.execute("DELETE FROM history")
-        conn.commit()
-        return []
-    clear.click(clear_history, None, chatbot)
+    # Clear button
+    clear.click(lambda: [], None, chatbot)
 
 if __name__ == "__main__":
     app.launch()