nikunjcepatel's picture
Update app.py
8658e51 verified
raw
history blame
909 Bytes
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()