Spaces:
Running
Running
| 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() |