Spaces:
Sleeping
Sleeping
import gradio as gr | |
import requests | |
import json | |
import os | |
# Retrieve the Open Router API Key from the Space secrets | |
API_KEY = os.getenv("OpenRounter_API_KEY") | |
def chat_with_openrouter(input_text): | |
response = requests.post( | |
url="https://openrouter.ai/api/v1/chat/completions", | |
headers={ | |
"Authorization": f"Bearer {API_KEY}" | |
}, | |
data=json.dumps({ | |
"model": "openai/gpt-4o-mini-2024-07-18", # Optional | |
"messages": [ | |
{"role": "user", "content": input_text} | |
] | |
"top_p": 1, | |
"temperature": 1, | |
"frequency_penalty": 0, | |
"presence_penalty": 0, | |
"repetition_penalty": 1, | |
"top_k": 0, | |
}) | |
) | |
return response.json().get("choices", [{}])[0].get("message", {}).get("content", "No response") | |
# Create Gradio interface | |
iface = gr.Interface(fn=generate_text, inputs="text", outputs="text") | |
iface.launch() | |