Update app.py

app.py

@@ -1,29 +1,33 @@
 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(prompt):
-    url = "https://api.openrouter.ai/v1/chat/completions"
-    headers = {"Authorization": "Bearer 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"}
-    data = {
-        "model": "openai/gpt-4o-mini-2024-07-18",
-        "messages": [{"role": "user", "content": prompt}],
+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,
-        "transforms": []
-    }
-    response = requests.post(url, headers=headers, json=data)
+        "top_k": 0,
+        })
+    )
     return response.json().get("choices", [{}])[0].get("message", {}).get("content", "No response")
 
-iface = gr.Interface(
-    fn=chat_with_openrouter,
-    inputs=gr.Textbox(lines=2, placeholder="Ask me anything"),
-    outputs="text",
-    title="Chat with OpenRouter",
-)
+# Create Gradio interface
+iface = gr.Interface(fn=generate_text, inputs="text", outputs="text")
 
 iface.launch()