Update app.py

app.py

@@ -34,12 +34,31 @@ def stream_chat_with_rag(
     return answer, fig
 
 
+    
+
+def heatmap(top_n):
+    fig = client.predict(
+    	query= message,
+		election_year=client_name,
+		api_name="/heatmap"
+    )
+    return fig
 
 # Create Gradio interface
 with gr.Blocks(title="Reddit Election Analysis") as demo:
     gr.Markdown("# Reddit Public sentiment & Social topic distribution ")
-
-
+    with gr.Row():
+        with gr.Column():
+            with gr.Row():
+                top_n = gr.Dropdown(choices= [1,2,3,4,5,6,7,8,9,10])
+            with gr.Row():
+                fresh_btn = gr.Button("Refresh Heatmap")
+        with gr.Column():
+            output_heatmap = gr.Plot(
+                label="Top Public sentiment & Social topic Heatmap",
+                container=True,  # Ensures the plot is contained within its area
+                elem_classes="heatmap-plot"  # Add a custom class for styling
+            )
 
     gr.Markdown("# Reddit Election Posts/Comments Analysis")
     gr.Markdown("Ask questions about election-related comments and posts")
@@ -86,8 +105,18 @@ with gr.Blocks(title="Reddit Election Analysis") as demo:
                 width: 100%;
                 margin: auto;
             }
+            .heatmap-plot {
+                min-height: 400px;
+                width: 100%;
+                margin: auto;
+            }
         </style>
     """)
+    fresh_btn.click(
+        fn=heatmap,
+        inputs=top_n,
+        outputs=output_heatmap
+    )
 
     # Update both outputs when submit is clicked
     submit_btn.click(
@@ -97,10 +126,5 @@ with gr.Blocks(title="Reddit Election Analysis") as demo:
     )
 
 
-
 if __name__ == "__main__":
-    demo.launch(share=True)
-    demo.launch(share_error=True)
-
-
-
+    demo.launch(share=True)