Update app.py

app.py

@@ -1,26 +1,23 @@
 import gradio as gr
 from transformers import pipeline
 
-# Load ChatGPT model (or use Hugging Face API)
-generator = pipeline("text-generation", model="gpt-3.5-turbo")
+# Load the AI model (OpenAI GPT or similar via transformers)
+generator = pipeline("text-generation", model="gpt2")
 
-# Function to generate code
-def generate_code(microcontroller, project, custom_input):
-    prompt = f"Generate {microcontroller} code for a {project} project. {custom_input}"
-    response = generator(prompt, max_length=500)
+def generate_code(prompt):
+    # Generate code based on the user's prompt
+    response = generator(prompt, max_length=200, num_return_sequences=1)
     return response[0]["generated_text"]
 
-# Gradio interface
-iface = gr.Interface(
+# Define the Gradio interface
+interface = gr.Interface(
     fn=generate_code,
-    inputs=[
-        gr.Dropdown(["Arduino", "ESP32", "STM32"], label="Microcontroller"),
-        gr.Dropdown(["LED Blinking", "Sensor Integration", "Motor Control"], label="Project Type"),
-        gr.Textbox(label="Custom Input (e.g., Pin Numbers, Delays)")
-    ],
-    outputs=gr.Textbox(label="Generated Code"),
+    inputs="text",
+    outputs="text",
     title="Embedded System Code Generator",
-    description="Automatically generate embedded system code for common microcontroller-based projects."
+    description="Enter a description of the microcontroller-based project, and this app will auto-generate code for you."
 )
 
-iface.launch()
+# Launch the app
+if __name__ == "__main__":
+    interface.launch()