"""
return browser_template.format(iframe_src)
def generate_website(query: str):
system_prompt = "You are a helpful web development assistant. When asked to create a website, respond with clean, modern HTML code wrapped in ```html``` tags. Include any necessary CSS within a style tag. Make the design responsive and visually appealing."
pipe = pipeline("text-generation", model="Qwen/Qwen2.5-0.5B")
input_text = f"{system_prompt}\n\nUser: {query}\nAssistant:"
try:
response = pipe(input_text, max_length=2048, num_return_sequences=1)[0]['generated_text']
assistant_response = response.split("Assistant:")[-1].strip()
clean_html = remove_code_block(assistant_response)
preview_html = send_to_preview(clean_html)
return preview_html
except Exception as e:
return send_to_preview(f"
Error: {str(e)}
")
with gr.Blocks(css=css) as demo:
with gr.Column(elem_classes="column"):
with gr.Row():
gr.HTML(value=svg, elem_classes="logo-row")
with gr.Row():
with gr.Column():
chatbot = gr.Chatbot(label="Chatbot", type="messages")
text_input = gr.Textbox(
elem_classes="textbox",
label="Describe the website you want to build",
placeholder="Example: Create a modern landing page with a hero section and contact form",
lines=3
)
with gr.Column(elem_classes="column"):
preview = gr.HTML(elem_id="preview-window", value=send_to_preview(""))
submit_btn = gr.Button("Generate", variant="primary", elem_id="submit-btn")
def on_submit(message, history):
history = history or []
print(message)
history.append({"role": "user", "content": message})
preview_html = generate_website(message)
return history, preview_html, gr.Textbox(value=None)
submit_btn.click(
fn=on_submit,
inputs=[text_input, chatbot],
outputs=[chatbot, preview, text_input],
)
if __name__ == "__main__":
demo.queue().launch()