Update app.py

app.py

@@ -1,32 +1,39 @@
+import requests
 import gradio as gr
 
-# Load Hugging Face Space that provides code generation (no token needed)
-generator = gr.Interface.load("huggingface/HuggingFaceH4/starchat-alpha")
+# Public inference endpoint
+API_URL = "https://huggingface.co/api-inference/v1/models/bigcode/santacoder"
 
 def generate_code(prompt, max_tokens):
-    try:
-        # Call the generator with just the prompt
-        result = generator(prompt)
-        
-        # Handle various return formats (tuple, list, dict, etc.)
-        if isinstance(result, (list, tuple)):
-            return result[0] if result else "No output received."
-        elif isinstance(result, dict):
-            return result.get("text", "No 'text' field in result.")
-        else:
-            return str(result)
-    except Exception as e:
-        return f"⚠️ Error generating code: {str(e)}"
+    payload = {
+        "inputs": prompt,
+        "parameters": {
+            "max_new_tokens": max_tokens,
+            "temperature": 0.7,
+            "do_sample": True,
+            "top_p": 0.95
+        }
+    }
 
+    headers = {}  # No token required
+    response = requests.post(API_URL, json=payload, headers=headers)
+
+    if response.status_code == 200:
+        result = response.json()
+        return result[0]["generated_text"]
+    else:
+        return f"⚠️ Error {response.status_code}: {response.text}"
+
+# Gradio interface
 interface = gr.Interface(
     fn=generate_code,
     inputs=[
-        gr.Textbox(lines=5, label="Code Prompt", placeholder="e.g., def factorial(n):"),
-        gr.Slider(minimum=20, maximum=300, step=10, label="Max Tokens (not used in this model)", value=100),
+        gr.Textbox(lines=5, label="Code Prompt", placeholder="e.g., def add(a, b):"),
+        gr.Slider(10, 200, step=10, label="Max Tokens", value=100),
     ],
     outputs=gr.Textbox(lines=10, label="Generated Code"),
     title="🧠 Code Generator (No Token Needed)",
-    description="This app uses a free Hugging Face-hosted model. No token or GPU needed!"
+    description="Uses Hugging Face hosted SantaCoder model. No login or GPU required."
 )
 
 if __name__ == "__main__":