Update app.py

app.py

@@ -9,12 +9,13 @@ HF_TOKEN = os.getenv("HF_TOKEN")  # Replace with your actual token if not using
 # Initialize the Gradio Client for the specified API
 client = Client("mangoesai/Elections_Comparison_Agent_V4", hf_token=HF_TOKEN)
 
-# client_name = ['2016 Election','2024 Election', 'Comparison two years']
+client_name = ['2016 Election','2024 Election', 'Comparison two years']
 
 
 
 def stream_chat_with_rag(
     message: str,
+    # history: list,
     client_name: str
 ):
     # print(f"Message: {message}")
@@ -32,36 +33,11 @@ def stream_chat_with_rag(
 
     return answer, fig
 
-def heatmap(top_n):
-    map = client.predict(
-        top_n = top_n,
-        api_name = "/heatmap"
-    )
-
-    return map
-
 
 
 # Create Gradio interface
 with gr.Blocks(title="Reddit Election Analysis") as demo:
-    gr.Markdown("# Reddit Public sentiment & Social topic Heatmap ")
-    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
-            )
-    fresh_btn.click(
-        fn=heatmap,
-        inputs=top_n,
-        outputs=output_heatmap
-    )
+    gr.Markdown("# Reddit Public sentiment & Social topic distribution ")
 
 
 
@@ -78,8 +54,7 @@ with gr.Blocks(title="Reddit Election Analysis") as demo:
 
             query_input = gr.Textbox(
                 label="Your Question",
-                placeholder="Ask about election comments or posts, like, Is there any comments don't like the election results?",
-                value = "Is there any comments don't like the election results"
+                placeholder="Ask about election comments or posts..."
             )
 
             submit_btn = gr.Button("Submit")
@@ -111,11 +86,6 @@ with gr.Blocks(title="Reddit Election Analysis") as demo:
                 width: 100%;
                 margin: auto;
             }
-            .heatmap-plot {
-                min-height: 400px;
-                width: 100%;
-                margin: auto;
-            }
         </style>
     """)
 
@@ -127,8 +97,9 @@ with gr.Blocks(title="Reddit Election Analysis") as demo:
     )
 
 
+
 if __name__ == "__main__":
-    demo.launch(share=True， show_error=True)
+    demo.launch(share=True)