Spaces:

parrotmaker
/

ai-canvas

Running

App Files Files Community

parrotmaker commited on Jun 17

Commit

299f2c4

verified ·

1 Parent(s): f80e4d3

Update app.py

Browse files

Files changed (1) hide show

app.py +73 -30

app.py CHANGED Viewed

@@ -1,41 +1,84 @@
 import gradio as gr
-# Placeholder assistant function
-def assistant_response(message):
-    return f"🧠 Assistant: You said '{message}'"
-# (Optional) Run code
-def run_code(code):
-    try:
-        exec_globals = {}
-        exec(code, exec_globals)
-        return "✅ Code executed successfully (no visible output)"
-    except Exception as e:
-        return f"❌ Error:\n{e}"
-with gr.Blocks() as demo:
-    with gr.Row():
-        # LEFT COLUMN - AI CHAT
-        with gr.Column(scale=1):
-            gr.Markdown("### 🤖 AI Assistant")
-            chat = gr.Chatbot()
-            msg = gr.Textbox(placeholder="Ask for help...", label="Your Message")
-            send = gr.Button("Send")
-            def handle_chat(user_message, history):
-                reply = assistant_response(user_message)
-                history.append((user_message, reply))
-                return "", history
-            send.click(handle_chat, [msg, chat], [msg, chat])
-        # RIGHT COLUMN - CODE EDITOR
         with gr.Column(scale=2):
-            gr.Markdown("### 💻 Code Canvas")
-            editor = gr.Code(label="Your Code Here", language="python")
-            run_btn = gr.Button("Run Code")
-            output = gr.Textbox(label="Output")
-            run_btn.click(run_code, inputs=editor, outputs=output)
 demo.launch()

 import gradio as gr
+from transformers import pipeline
+import zipfile
+import io
+# Load AI assistant (can change model to faster one if needed)
+assistant = pipeline("text-generation", model="mistralai/Mixtral-8x7B-Instruct-v0.1", max_new_tokens=1024)
+def generate_code(prompt):
+    formatted = f"Create a full website. Include:\n[HTML]\n...HTML here...\n[CSS]\n...CSS here...\n[JS]\n...JS here...\nUser Prompt: {prompt}"
+    response = assistant(formatted)[0]["generated_text"]
+    # Try to split out HTML, CSS, JS sections
+    html = css = js = ""
+    if "[HTML]" in response and "[CSS]" in response and "[JS]" in response:
+        try:
+            html = response.split("[HTML]")[1].split("[CSS]")[0].strip()
+            css = response.split("[CSS]")[1].split("[JS]")[0].strip()
+            js = response.split("[JS]")[1].strip()
+        except:
+            html = response
+    else:
+        html = response
+    return html, css, js, f"✅ AI generated your website!"
+def combine_code(html, css, js):
+    return f"""
+<!DOCTYPE html>
+<html>
+<head>
+  <style>{css}</style>
+</head>
+<body>
+{html}
+<script>{js}</script>
+</body>
+</html>
+"""
+def export_zip(html, css, js):
+    in_memory = io.BytesIO()
+    with zipfile.ZipFile(in_memory, 'w') as zf:
+        zf.writestr("index.html", f"<!DOCTYPE html><html><head><link rel='stylesheet' href='style.css'></head><body>\n{html}\n<script src='script.js'></script></body></html>")
+        zf.writestr("style.css", css)
+        zf.writestr("script.js", js)
+    in_memory.seek(0)
+    return (in_memory, "website.zip")
+with gr.Blocks(css="""
+body { font-family: 'Courier New', monospace; background-color: #1e1e1e; color: white; }
+.gr-button { background-color: #007acc; color: white; border: none; }
+.gr-code textarea { background-color: #252526; color: white; font-size: 14px; }
+.gr-textbox textarea { background-color: #1e1e1e; color: white; }
+.gr-markdown h2 { color: #61dafb; }
+""") as demo:
+    with gr.Row():
+        with gr.Column(scale=1):
+            gr.Markdown("## 🤖 AI Assistant")
+            prompt = gr.Textbox(label="Describe the website you want", placeholder="e.g., A landing page for a pet adoption service")
+            generate_btn = gr.Button("✨ Generate Website")
+            status = gr.Textbox(label="Status")
         with gr.Column(scale=2):
+            gr.Markdown("## 💻 Code Editor (HTML / CSS / JS)")
+            with gr.Tabs():
+                with gr.Tab("HTML"):
+                    html_editor = gr.Code(language="html", label="HTML")
+                with gr.Tab("CSS"):
+                    css_editor = gr.Code(language="css", label="CSS")
+                with gr.Tab("JavaScript"):
+                    js_editor = gr.Code(language="javascript", label="JavaScript")
+            run_btn = gr.Button("▶️ Preview")
+            preview = gr.HTML()
+            zip_btn = gr.Button("📦 Export as ZIP")
+            zip_file = gr.File()
+    # Button Actions
+    generate_btn.click(generate_code, inputs=prompt, outputs=[html_editor, css_editor, js_editor, status])
+    run_btn.click(combine_code, inputs=[html_editor, css_editor, js_editor], outputs=preview)
+    zip_btn.click(export_zip, inputs=[html_editor, css_editor, js_editor], outputs=zip_file)
 demo.launch()