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

# Create Gradio interface
iface = gr.Interface(fn=generate_text, inputs="text", outputs="text")

iface.launch()