|
import gradio as gr |
|
import requests |
|
import json |
|
import os |
|
|
|
|
|
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", |
|
"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") |
|
|
|
|
|
iface = gr.Interface(fn=generate_text, inputs="text", outputs="text") |
|
|
|
iface.launch() |
|
|