Update app.py

app.py

@@ -1,7 +1,18 @@
 import gradio as gr
+import requests
+
+API_URL = "https://api-inference.huggingface.co/models/bigcode/starcoder"
+import os
+headers = {"Authorization": f"Bearer {os.environ['HF_TOKEN']}"}
 
 def coder_bot(instruction):
-    return "Your AI Coder Bot received: " + instruction + "\n(Real code generation coming in next step!)"
+    payload = {"inputs": instruction}
+    response = requests.post(API_URL, headers=headers, json=payload)
+    result = response.json()
+    if isinstance(result, list) and "generated_text" in result[0]:
+        return result[0]["generated_text"]
+    else:
+        return "Something went wrong or model is loading. Please try again in a few seconds."
 
 demo = gr.Interface(fn=coder_bot, inputs="text", outputs="text")
 demo.launch()