Update app.py

app.py

@@ -1,22 +1,29 @@
 import gradio as gr
 from transformers import pipeline
 
-# Load model (you can change this to your own)
-generator = pipeline("text2text-generation", model="google/flan-t5-base")
+# Load code-aware model
+generator = pipeline("text-generation", model="microsoft/phi-1_5")
 
-# Function to generate code from prompt
-def generate_code(prompt):
-    result = generator(prompt, max_new_tokens=100)[0]["generated_text"]
+# Code assistant function
+def generate_from_code(prompt):
+    result = generator(
+        prompt,
+        max_new_tokens=256,
+        do_sample=True,
+        temperature=0.7,
+        pad_token_id=generator.tokenizer.eos_token_id
+    )[0]["generated_text"]
+    
     return result.strip()
 
-# Gradio UI and API Interface
+# Gradio app with UI and API mode
 iface = gr.Interface(
-    fn=generate_code,
-    inputs=gr.Textbox(label="Enter prompt (e.g., Describe code)"),
-    outputs=gr.Textbox(label="Generated Code"),
-    title="Code Generator AI",
-    description="This app generates code based on your prompt. You can use it from the browser or call it via API."
+    fn=generate_from_code,
+    inputs=gr.Textbox(lines=10, placeholder="Paste code or describe what to do...", label="Prompt / Code"),
+    outputs=gr.Textbox(label="AI Output"),
+    title="AI Code Helper",
+    description="Give a prompt, code snippet, or question. Example: 'Explain this code', 'Complete this function', or 'Fix the bug'."
 )
 
-# Launch Space (UI + API mode enabled)
-iface.launch()
+# Enable public API
+iface.launch(share=True)