ZeppFusion / app.py
Xyro123's picture
Update app.py
4a7d211 verified
raw
history blame
1.11 kB
import gradio as gr
import openai
import requests
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:
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()