Better Output

app.py

@@ -84,8 +84,14 @@ NEVER USE HASHTAGS.
     )
     r = json.loads(completion.choices[0].message.function_call.arguments)
     tweets = [{'type': tweet_type.split('_')[0], 'text': t['content']} for tweet_type in r for t in r[tweet_type]]
-
-    return pd.DataFrame(tweets)
+    df = pd.DataFrame(tweets)
+    grouped = df.groupby('type')['text'].apply(list).reset_index()
+    markdown = ""
+    for _, row in grouped.iterrows():
+        markdown += f"- **{row['type']}**\n"
+        for text in row['text']:
+            markdown += f"  - {text}\n"
+    return markdown
 
 
 
@@ -94,8 +100,8 @@ with gr.Blocks() as app:
     api_key_input = gr.Textbox(label="Password", placeholder="Enter the super secret password")
     abstract_input = gr.Textbox(label="Article Abstract", placeholder="Enter the article abstract here")
     generate_button = gr.Button("Generate Tweets")
-    output_dataframe = gr.Dataframe()
+    output_markdown = gr.Markdown()
     
-    generate_button.click(fn=write_tweets, inputs=[api_key_input, abstract_input], outputs=output_dataframe)
+    generate_button.click(fn=write_tweets, inputs=[api_key_input, abstract_input], outputs=output_markdown)
 
 app.launch()