import gradio as gr
from transformers import pipeline

# Load the AI model (OpenAI GPT or similar via transformers)
generator = pipeline("text-generation", model="gpt2")

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"]

# Define the Gradio interface
interface = gr.Interface(
    fn=generate_code,
    inputs="text",
    outputs="text",
    title="Embedded System Code Generator",
    description="Enter a description of the microcontroller-based project, and this app will auto-generate code for you."
)

# Launch the app
if __name__ == "__main__":
    interface.launch()