Update app.py

app.py

@@ -1,18 +1,25 @@
 import gradio as gr
 import requests
+import os
 
 API_URL = "https://api-inference.huggingface.co/models/bigcode/starcoder"
-import os
 headers = {"Authorization": f"Bearer {os.environ['HF_TOKEN']}"}
 
-def coder_bot(instruction):
-    payload = {"inputs": instruction}
+def generate_code(prompt):
+    payload = {
+        "inputs": prompt,
+        "parameters": {"max_new_tokens": 300}
+    }
+    
     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."
+    
+    try:
+        generated_text = response.json()[0]["generated_text"]
+    except Exception as e:
+        generated_text = f"Error: {e}\nRaw Response: {response.text}"
+    
+    return generated_text
+
+demo = gr.Interface(fn=generate_code, inputs="text", outputs="text", title="Super Coder Bot", description="Enter instructions in plain English. Get real, production-ready code.")
 
-demo = gr.Interface(fn=coder_bot, inputs="text", outputs="text")
 demo.launch()