Create app.py

app.py

@@ -0,0 +1,21 @@
+import gradio as gr
+from transformers import pipeline
+
+# Using a lightweight model to stay within free tier limits
+generator = pipeline("text-generation", model="Salesforce/codegen2-1B", max_new_tokens=1024)
+
+def generate_website_code(prompt):
+    system_prompt = (
+        f"Create a fully responsive website with HTML, CSS, and JavaScript. "
+        f"Use these requirements: {prompt}.\n\n"
+        f"Return full code with all parts embedded."
+    )
+    result = generator(system_prompt)[0]["generated_text"]
+    return result
+
+gr.Interface(
+    fn=generate_website_code,
+    inputs=gr.Textbox(lines=5, label="Describe the website you want (theme, brand, images, colors)"),
+    outputs="text",
+    title="AI Website Generator"
+).launch()