Update app.py

app.py

@@ -59,19 +59,20 @@ def predict(message, history):
     return gpt_response.content
     
 def chat_function(message, history, year):
-    history_langchain_format = []
-    for msg in history:
-        if msg['role'] == "user":
-            history_langchain_format.append(HumanMessage(content=msg['content']))
-        elif msg['role'] == "assistant":
-            history_langchain_format.append(AIMessage(content=msg['content']))
-    history_langchain_format.append(HumanMessage(content=message))
-    rag_response = stream_chat_with_rag(history_langchain_format,year)[0]
+    # history_langchain_format = []
+    # for msg in history:
+    #     if msg['role'] == "user":
+    #         history_langchain_format.append(HumanMessage(content=msg['content']))
+    #     elif msg['role'] == "assistant":
+    #         history_langchain_format.append(AIMessage(content=msg['content']))
+    # history_langchain_format.append(HumanMessage(content=message))
+    # rag_response = stream_chat_with_rag(history_langchain_format,year)[0]
     
-    # response = f"Year selected: {year}. Here's a response."
-    # answer = stream_chat_with_rag(user_message, year)[0]
-    # chat_history.append((user_message, response +"\n"+ answer))
-    return rag_response
+    response = f"Year selected: {year}. Here's a response."
+    answer = stream_chat_with_rag(message, year)[0]
+    history.append((message, response +"\n"+ answer))
+    
+    return answer
 
 chatbot_state = gr.State([])
 
@@ -204,12 +205,14 @@ with gr.Blocks(title="Reddit Election Analysis") as demo:
     gr.Markdown("Ask questions about election-related comments and posts")
 
     with gr.Row():
-        # with gr.Column():
-        #     year_selector = gr.Radio(
-        #         choices=["2016 Election", "2024 Election", "Comparison two years"],
-        #         label="Select Election Year",
-        #         value="2016 Election"
-        #     )
+        with gr.Column():
+            year_selector = gr.Radio(
+                choices=["2016 Election", "2024 Election", "Comparison two years"],
+                label="Select Election Year",
+                value="2016 Election"
+            )
+            slider = gr.Slider(50, 500, render=False, label= "Tokens")
+
 
         #     query_input = gr.Textbox(
         #         label="Your Question",
@@ -233,13 +236,8 @@ with gr.Blocks(title="Reddit Election Analysis") as demo:
         with gr.Column():
             chatbot = gr.ChatInterface(chat_function, 
                                        type="messages", 
-                                       addtional_inputs = [
-                                            year_selector = gr.Radio(
-                                                choices=["2016 Election", "2024 Election", "Comparison two years"],
-                                                label="Select Election Year",
-                                                value="2016 Election"
-                                            )
-                                       ])
+                                       addtional_inputs = [year_selector]
+                                      )
             
     gr.Markdown("## Top works of the relevant Q&A")
     # with gr.Row():