Update app.py

app.py

@@ -22,9 +22,12 @@ def article_selection(sentiment):
     if sentiment == "Positive":
         predictions = negative_preds
         predictions_df_url0 = predictions['Url'].iloc[0]
+        predictions_df_head0 = predictions['Headlines_string'].iloc[0]
         predictions_df_url1 = predictions['Url'].iloc[1]
+        predictions_df_head1 = predictions['Headlines_string'].iloc[1]
         predictions_df_url2 = predictions['Url'].iloc[2]
-        return predictions_df_url0, predictions_df_url1, predictions_df_url2
+        predictions_df_head2 = predictions['Headlines_string'].iloc[2]
+        return [[predictions_df_head0 predictions_df_url0], [predictions_df_head1 predictions_df_url1], [predictions_df_head2 predictions_df_url2] 
     elif sentiment == "Negative":
         predictions = negative_preds
         predictions_df_url0 = predictions['Url'].iloc[0]
@@ -63,7 +66,8 @@ suggestion_demo = gr.Interface(
     fn=article_selection,
     title = 'Recommending News Articles',
     inputs = gr.Dropdown(["Positive","Negative","Neutral"], label="What type of news articles would you like recommended?"),
-    outputs = [gr.Textbox(label="Recommended News Articles (1/3)"),gr.Textbox(label="Recommended News Articles (2/3)"),gr.Textbox(label="Recommended News Articles (3/3)")],
+    outputs = "dataframe",
+    #outputs = [gr.Textbox(label="Recommended News Articles (1/3)"),gr.Textbox(label="Recommended News Articles (2/3)"),gr.Textbox(label="Recommended News Articles (3/3)")],
     description = description1
 )