Update app.py

app.py

@@ -167,8 +167,7 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
     )
 
     prompt = PromptTemplate(
-    #input_variables=["input", "agent_scratchpad", "chat_history", "tool_names"],
-    input_variables=["input", "agent_scratchpad", "tool_names"],
+    input_variables=["input", "agent_scratchpad", "chat_history", "tool_names"],
     template="""
     You are a smart and helpful AI Agent/Assistant that excels at fact-based reasoning. You are allowed and encouraged to use one or more tools as needed to answer complex questions and perform tasks.
     It is CRUCIAL that you ALWAYS follow the exact format below. Do not deviate.
@@ -305,6 +304,8 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
     Final Answer: LUX
 
     ---
+    Previous conversation history:
+    {chat_history}
 
     New input: {input}
     ---
@@ -312,8 +313,7 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
     """
     )
 
-    #summary_memory = ConversationSummaryMemory(llm=llm_client, memory_key="chat_history")
-    summary_memory = None
+    summary_memory = ConversationSummaryMemory(llm=llm_client, memory_key="chat_history")
 
 
     # Initialize gemini model with streaming enabled
@@ -384,7 +384,7 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
             full_question_for_agent += f"\n\nAttachment '{file_name}' available at EXACT URL: {attachment_url}"
             print(f"Running agent on task {task_id}: {full_question_for_agent}",flush=True)
         
-        
+        '''
         allowed_ids = {
             #"cca530fc-4052-43b2-b130-b30968d8aa44",
             #"a1e91b78-d3d8-4675-bb8d-62741b4b68a6",
@@ -397,7 +397,7 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
         }
         if task_id not in allowed_ids:
             continue
-        
+        '''
         try:
             submitted_answer = agent(full_question_for_agent)
             answers_payload.append({"task_id": task_id, "submitted_answer": submitted_answer})