Update app.py

app.py

@@ -47,35 +47,55 @@ def stream_chat_with_rag(
 
     return answer
 
-CSS = """
-# chat-container {
-    height: 100vh;
-}
-"""
-    
-
 
-# Title for the application
-TITLE = "<h1 style='text-align:center;'>Reddit Election Q&A agent v0.2</h1>"
-
-# Create the Gradio Blocks interface
-with gr.Blocks(css=CSS) as demo:
-    gr.HTML(TITLE)
-    with gr.Tab("Chat"):
-        chatbot = gr.Chatbot()  # Create a chatbot interface
-        chat_interface = gr.ChatInterface(
-            fn=stream_chat_with_rag,
-            chatbot=chatbot,
-            additional_inputs_accordion=gr.Accordion(label="⚙️ Parameters", open=False, render=False),
-            additional_inputs=[
-                gr.Dropdown(client_name,value="2016 Election",label="Select Election year", render=False,allow_custom_value=True)
-            ],
-        )
 
+# Create Gradio interface
+with gr.Blocks(title="Reddit Election Comments Analysis") as demo:
+    gr.Markdown("# Reddit Election 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
+            )
+            
+            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")
+        
+        with gr.Column():
+            output = gr.Textbox(
+                label="Response",
+                lines=20
+            )
+    
+    # Update submit button to include year selection
+    submit_btn.click(
+        fn=stream_chat_with_rag,
+        inputs=[query_input, year_selector],
+        outputs=output
+    )
+    
+    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?
+     """)
 
-# Launch the app
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch(share=True)