Update app.py

app.py

@@ -32,7 +32,10 @@ def process(user_input):
     log = "\n\n".join([f"{msg.type.upper()}: {msg.content}" for msg in final_state["messages"]])
 
     html_file, log_file, zip_file = save_files(html, log)
-    preview = f"<iframe src='file/{html_file}' width='100%' height='500px' style='border:1px solid #ccc;'></iframe>"
+
+    # Escape quotes for safe iframe embedding
+    escaped_html = html.replace('"', "&quot;")
+    preview = f"<iframe srcdoc='{escaped_html}' width='100%' height='600px' style='border:1px solid #ccc;'></iframe>"
 
     return preview, html, log, html_file, log_file, zip_file
 
@@ -42,11 +45,8 @@ with gr.Blocks() as demo:
     run_btn = gr.Button("Run Agents")
 
     gr.Markdown("### 🔍 Website Preview")
+    preview_html = gr.HTML()
 
-    escaped_html = html.replace('"', '&quot;')
-    preview_html = gr.HTML(value=f"<iframe srcdoc='{escaped_html}' width='100%' height='600px'></iframe>")
-
-    
     gr.Markdown("### 🖥️ Generated HTML Code")
     html_output = gr.Textbox(lines=15, label="Final HTML")
 
@@ -65,4 +65,4 @@ with gr.Blocks() as demo:
     )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()