Upload app.py

app.py

@@ -0,0 +1,59 @@
+import os
+import gradio as gr
+from openai import OpenAI
+
+# --- Initialize client securely ---
+HF_TOKEN = os.getenv("HF_TOKEN")
+
+if not HF_TOKEN:
+    raise ValueError("❌ HF_TOKEN not found. Please set it in your Hugging Face Space secrets.")
+
+client = OpenAI(
+    base_url="https://router.huggingface.co/v1",
+    api_key=HF_TOKEN,
+)
+
+# --- Chat handler ---
+def chat_with_model(message, history):
+    # Build messages list safely
+    messages = []
+
+    if history:
+        for msg in history:
+            # Handle both dict and tuple formats
+            if isinstance(msg, dict):
+                # Keep only allowed keys
+                messages.append({
+                    "role": msg.get("role", "user"),
+                    "content": msg.get("content", "")
+                })
+            elif isinstance(msg, (list, tuple)) and len(msg) == 2:
+                messages.append({"role": "user", "content": msg[0]})
+                messages.append({"role": "assistant", "content": msg[1]})
+
+    # Add latest user message
+    messages.append({"role": "user", "content": message})
+
+    try:
+        response = client.chat.completions.create(
+            model="openai/gpt-oss-120b:fireworks-ai",
+            messages=messages,
+            temperature=0.7,
+        )
+        reply = response.choices[0].message.content
+    except Exception as e:
+        reply = f"⚠️ Error: {str(e)}"
+
+    return reply
+
+# --- Gradio Interface ---
+chatbot_ui = gr.ChatInterface(
+    fn=chat_with_model,
+    title="🧠 GPT-OSS 120B (Fireworks)",
+    description="Chat with the OSS 120B model hosted via Hugging Face router.",
+    examples=["Hello!", "Tell me a joke.", "Explain AI in simple terms."],
+    type="messages",  # required
+)
+
+if __name__ == "__main__":
+    chatbot_ui.launch()