Update app.py

app.py

@@ -1,31 +1,17 @@
 import gradio as gr
 from utils.langgraph_pipeline import run_pipeline_and_save
-from utils.zip_output import zip_output
 
 def handle_run(prompt):
     chat_log, final_output = run_pipeline_and_save(prompt)
-    return (
-        [(f"{agent}", msg) for step in chat_log for agent, msg in step.items()],
-        "output/index.html",
-        "output/agent_log.json",
-        zip_output()
-    )
+    return chat_log, final_output
 
 with gr.Blocks() as demo:
-    gr.Markdown("# 🔗 Multi-Agent Collaboration with LangGraph")
+    gr.Markdown("# 🔧 Multi-Agent UI Generator")
+    input_box = gr.Textbox(lines=6, label="Enter your product idea prompt")
+    run_btn = gr.Button("Generate")
+    chatbox = gr.Chatbot(label="Agent Conversation Log", type="messages")
+    file_output = gr.File(label="Download UI ZIP")
 
-    prompt_input = gr.Textbox(label="Enter your prompt", lines=4, placeholder="Describe your product idea...")
-    submit_btn = gr.Button("Run Agents")
-
-    chatbox = gr.Chatbot(label="Agent Conversation Log")
-    html_out = gr.File(label="index.html")
-    log_out = gr.File(label="Agent Log")
-    zip_out = gr.File(label="ZIP Output")
-
-    submit_btn.click(
-        fn=handle_run,
-        inputs=prompt_input,
-        outputs=[chatbox, html_out, log_out, zip_out]
-    )
+    run_btn.click(fn=handle_run, inputs=input_box, outputs=[chatbox, file_output])
 
 demo.launch()