Update app.py

app.py

@@ -1,17 +1,31 @@
 import gradio as gr
+import openai
+import requests
 
-def respond(message, chat_history):
-    if chat_history is None:
-        chat_history = []
-    # Энгийн reply
-    reply = f"Чи бичсэн: {message}"
-    chat_history.append((message, reply))
-    return chat_history, chat_history
+OPENAI_API_KEY = "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"
+GEMINI_API_KEY = "AIzaSyCOREn4MTSkCPRK_0BF-CY6qpFEmMda64E"
+
+def fusion_response(message, mode="Fusion"):
+    chatgpt_resp = openai.ChatCompletion.create(
+        model="gpt-4",
+        messages=[{"role":"user","content":message}]
+    )["choices"][0]["message"]["content"]
+
+    gemini_resp = requests.post(
+        "https://gemini-api-url",
+        headers={"Authorization": f"Bearer {GEMINI_API_KEY}"},
+        json={"prompt": message}
+    ).json()["reply"]
+
+    if mode=="ChatGPT": return chatgpt_resp
+    if mode=="Gemini": return gemini_resp
+    # Fusion mode
+    return f"ChatGPT: {chatgpt_resp}\nGemini: {gemini_resp}"
 
 with gr.Blocks() as demo:
-    chatbot = gr.Chatbot()
-    msg = gr.Textbox(label="Message")
-    state = gr.State()
-    msg.submit(respond, [msg, state], [chatbot, state])
+    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)
 
 demo.launch()