Update app.py

app.py

@@ -3,34 +3,42 @@ from huggingface_hub import InferenceClient
 import os
 
 # Configuration
-API_KEY = os.getenv("API_KEY", "your-secret-key")  # Set in Space Secrets
-MODEL_NAME = "HuggingFaceH4/zephyr-7b-beta"
-SYSTEM_PROMPT = "You're a helpful cooking assistant. Provide detailed recipes."
+API_KEY = os.getenv("API_KEY", "your-secret-key-123")  # Set in Settings → Secrets
+MODEL = "HuggingFaceH4/zephyr-7b-beta"
 
-client = InferenceClient(MODEL_NAME)
+client = InferenceClient(MODEL)
 
-def generate_response(message: str, api_key: str):
+def chat_api(message: str, api_key: str):
     """Secure API endpoint"""
     if api_key != API_KEY:
         return {"error": "Invalid API key"}
     
-    messages = [
-        {"role": "system", "content": SYSTEM_PROMPT},
-        {"role": "user", "content": message}
-    ]
-    
-    try:
-        return client.chat_completion(messages, max_tokens=500)
-    except Exception as e:
-        return {"error": str(e)}
+    response = client.chat_completion(
+        messages=[{"role": "user", "content": message}],
+        max_tokens=500,
+        stream=False
+    )
+    return {"response": response.choices[0].message.content}
 
-# Minimal UI for testing
-demo = gr.Interface(
-    fn=lambda msg: generate_response(msg, API_KEY),
-    inputs="text",
-    outputs="json",
-    title="🍳 Recipe API Gateway",
-    description=f"API Key: `{API_KEY}`"
-)
+# Minimal UI with API docs
+with gr.Blocks() as demo:
+    gr.Markdown("## 🍳 Recipe Chatbot API Gateway")
+    with gr.Tab("Chat"):
+        gr.ChatInterface(lambda msg: chat_api(msg, API_KEY)["response"])
+    with gr.Tab("API Docs"):
+        gr.Markdown(f"""
+        ### API Endpoint
+        ```
+        POST https://sarthak413-chatbot.hf.space/api/predict
+        ```
+        **Required JSON:**
+        ```json
+        {{
+          "data": [
+            "your message",
+            "your-api-key-here"
+          ]
+        }}
+        """)
 
-demo.launch()
+demo.launch(show_api=True)