Update app.py

app.py

@@ -34,7 +34,8 @@ def stream_chat_with_rag(
     print("top works from API:")
     print(fig)
 
-    return answer, fig
+    # return answer, fig
+    return answer
 
 
     
@@ -44,6 +45,8 @@ def heatmap(top_n):
     	top_n= top_n,
 		api_name="/get_heatmap_pivot_table"
     )
+    print(pivot_table)
+    print(type(pivot_table))
     
     plt.figure(figsize=(10, 8))
     sns.heatmap(pivot_table,
@@ -135,10 +138,15 @@ with gr.Blocks(title="Reddit Election Analysis") as demo:
     )
 
     # Update both outputs when submit is clicked
+    # submit_btn.click(
+    #     fn=stream_chat_with_rag,
+    #     inputs=[query_input, year_selector],
+    #     outputs=[output_text, output_plot]
+    # )
     submit_btn.click(
         fn=stream_chat_with_rag,
         inputs=[query_input, year_selector],
-        outputs=[output_text, output_plot]
+        outputs=output_text
     )