Update app.py

app.py

@@ -1,35 +1,21 @@
 import gradio as gr
-from agent import graph, State
+from agent import workflow
 
-def generate_post(topic, platform):
-    initial_state: State = {
-        "topic": topic,
-        "platform": platform
-    }
-    final_state = graph.run(initial_state)
-
-    post = final_state.get("post", "No post generated")
-    engagement = final_state.get("engagement_score", "N/A")
-    tone = final_state.get("tone_score", "N/A")
-    clarity = final_state.get("clarity_score", "N/A")
-    images = "\n".join(final_state.get("image_url", [])) or "No images found"
-
-    return post, engagement, tone, clarity, images
+def generate(topic, platform):
+    post, engagement, tone, clarity = workflow(topic, platform)
+    return post, engagement, tone, clarity
 
 with gr.Blocks() as demo:
-    gr.Markdown("# Social Media Post Generator for B2B Banking")
-
-    topic = gr.Textbox(label="Topic", value="digital transformation in banking")
-    platform = gr.Dropdown(choices=["LinkedIn", "Twitter", "Facebook"], value="LinkedIn", label="Platform")
-
-    btn = gr.Button("Generate Post")
-    output_post = gr.Textbox(label="Generated Post", lines=10)
-    output_engagement = gr.Textbox(label="Engagement Score")
-    output_tone = gr.Textbox(label="Tone Score")
-    output_clarity = gr.Textbox(label="Clarity Score")
-    output_images = gr.Textbox(label="Image URLs")
-
-    btn.click(generate_post, inputs=[topic, platform],
-              outputs=[output_post, output_engagement, output_tone, output_clarity, output_images])
-
-demo.launch()
+    gr.Markdown("# Social Media Post Generator")
+    topic_input = gr.Textbox(label="Topic", placeholder="Enter topic here")
+    platform_input = gr.Textbox(label="Platform", placeholder="e.g., LinkedIn, Twitter")
+    generate_btn = gr.Button("Generate Post")
+    post_output = gr.Textbox(label="Generated Post", lines=8)
+    engagement_output = gr.Textbox(label="Engagement Score")
+    tone_output = gr.Textbox(label="Tone Score")
+    clarity_output = gr.Textbox(label="Clarity Score")
+
+    generate_btn.click(fn=generate, inputs=[topic_input, platform_input], outputs=[post_output, engagement_output, tone_output, clarity_output])
+
+if __name__ == "__main__":
+    demo.launch()