File size: 909 Bytes
66ff3e9
 
8658e51
b944409
 
 
 
66ff3e9
8658e51
 
 
 
 
 
 
 
 
 
 
c307a1d
 
8658e51
 
c307a1d
8658e51
 
 
66ff3e9
 
8658e51
 
66ff3e9
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
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()