Update app.py

app.py

@@ -1,20 +1,32 @@
 import gradio as gr
 
-# Load a pre-hosted code-generation model Space (no token needed)
+# Load Hugging Face Space that provides code generation (no token needed)
 generator = gr.Interface.load("huggingface/HuggingFaceH4/starchat-alpha")
 
 def generate_code(prompt, max_tokens):
-    return generator(prompt, max_tokens)[0]
+    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)}"
 
 interface = gr.Interface(
     fn=generate_code,
     inputs=[
-        gr.Textbox(label="Code Prompt", lines=5, placeholder="e.g., def add(a, b):"),
-        gr.Slider(20, 300, step=10, label="Max Tokens", value=100),
+        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),
     ],
-    outputs=gr.Textbox(label="Generated Code"),
-    title="🧠 Code Generator (No Token Required)",
-    description="Powered by Hugging Face hosted model. No API key needed."
+    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!"
 )
 
 if __name__ == "__main__":