Update app.py

app.py

@@ -7,7 +7,7 @@ import os
 HF_TOKEN = os.getenv("HF_TOKEN")  # Replace with your actual token if not using an environment variable
 
 # Initialize the Gradio Client for the specified API
-client = Client("mangoesai/Elections_Comparing_Agent_V2", hf_token=HF_TOKEN)
+client = Client("mangoesai/Elections_Comparison_Agent_V3", hf_token=HF_TOKEN)
 
 client_name = ['2016 Election','2024 Election', 'Comparison two years']
 
@@ -19,22 +19,8 @@ def stream_chat_with_rag(
     client_name: str
 ):
     # print(f"Message: {message}")
-    # print(f"History: {history}")
-
-    # # Build the conversation prompt including system prompt and history
-    # conversation = f"For Client: {client_name}\n"
-    
-    # # Add previous conversation history
-    # for user_input, assistant_response in history:
-    #     conversation += f"User: {user_input}\nAssistant: {assistant_response}\n"
-    
-    # # Add the current user message
-    # conversation += f"User: {message}\nAssistant:"
-
-    # # # Call the API with the user's process_query
-    # question = message
     #answer = client.predict(question=question, api_name="/run_graph")
-    answer = client.predict(
+    answer, fig = client.predict(
     	query= message,
 		election_year=client_name,
 		api_name="/process_query"
@@ -45,54 +31,72 @@ def stream_chat_with_rag(
     print(answer)
 
 
-    return answer
+    return answer, fig
 
 
 
 # Create Gradio interface
-with gr.Blocks(title="Reddit Election Comments Analysis") as demo:
-    gr.Markdown("# Reddit Election Comments Analysis")
+with gr.Blocks(title="Reddit Election Analysis") as demo:
+    gr.Markdown("# Reddit Public sentiment & Social topic distribution ")
+
+
+
+    gr.Markdown("# Reddit Election Posts/Comments Analysis")
     gr.Markdown("Ask questions about election-related comments and posts")
-    
+
     with gr.Row():
         with gr.Column():
-            # Add election year selector
             year_selector = gr.Radio(
                 choices=["2016 Election", "2024 Election", "Comparison two years"],
                 label="Select Election Year",
-                value="2016 Election"  # Default value
+                value="2016 Election"
             )
-            
+
             query_input = gr.Textbox(
                 label="Your Question",
                 placeholder="Ask about election comments or posts..."
             )
-            # context_input = gr.Textbox(
-            #     label="Context (Optional)",
-            #     value = "Looking for discussions about the election results in 2016" #default value
-            # )
+
             submit_btn = gr.Button("Submit")
-        
+
+            gr.Markdown("""
+            ## Example Questions:
+            - Is there any comments don't like the election results
+            - Summarize the main discussions about voting process
+            - What are the common opinions about candidates?
+            """)
         with gr.Column():
-            output = gr.Textbox(
+            output_text = gr.Textbox(
                 label="Response",
                 lines=20
             )
-    
-    # Update submit button to include year selection
+
+    with gr.Row():
+        output_plot = gr.Plot(
+            label="Topic Distribution",
+            container=True,  # Ensures the plot is contained within its area
+            elem_classes="topic-plot"  # Add a custom class for styling
+        )
+
+    # Add custom CSS to ensure proper plot sizing
+    gr.HTML("""
+        <style>
+            .topic-plot {
+                min-height: 600px;
+                width: 100%;
+                margin: auto;
+            }
+        </style>
+    """)
+
+    # Update both outputs when submit is clicked
     submit_btn.click(
         fn=stream_chat_with_rag,
         inputs=[query_input, year_selector],
-        outputs=output
+        outputs=[output_text, output_plot]
     )
-    
-    gr.Markdown("""
-    ## Example Questions:
-    - Is there any comments don't like the election results
-    - Summarize the main discussions about voting process
-    - What are the common opinions about candidates?
-    - How have people's attitudes toward the Republican Party changed in the past two years?
-     """)
+
+
 
 if __name__ == "__main__":
     demo.launch(share=True)