Update app.py

app.py

@@ -243,11 +243,12 @@ def Web_Search(history):
 
     msg=HumanMessage(content=augmented_prompt)
     messages2.append(msg)
-    response = chat_model.invoke(messages2)
-    messages2.append(AIMessage(content=response.content))
 
     if len(messages2) >= 8:
-      messages2 = messages2[-8:]
+        messages2 = messages2[-8:]
+        
+    response = chat_model.invoke(messages2)
+    messages2.append(AIMessage(content=response.content))
 
     history[-1][1] = ""
     for character in response.content:
@@ -270,13 +271,13 @@ def Chart_Generator(history):
     
             prompt = HumanMessage(content=message_with_description)
             messages3.append(prompt)
+
+            if len(messages3) >= 8:
+                messages3 = messages3[-8:]
     
             response = chat_model.invoke(messages3)
             messages3.append(AIMessage(content=response.content))
     
-            if len(messages3) >= 8:
-                messages3 = messages3[-8:]
-    
             combined_content = f'{image_html}<br>{response.content}'
         else:
             response_text = "Can't generate this image. Please provide valid chart details."
@@ -284,12 +285,12 @@ def Chart_Generator(history):
     else:
         prompt = HumanMessage(content=message)
         messages3.append(prompt)
+
+        if len(messages3) >= 8:
+            messages3 = messages3[-8:]
     
         response = chat_model.invoke(messages3)
         messages3.append(AIMessage(content=response.content))
-    
-        if len(messages3) >= 8:
-            messages3 = messages3[-8:]
 
         combined_content=response.content
         
@@ -333,12 +334,13 @@ def Link_Scratch(history):
             """
             message = HumanMessage(content=augmented_prompt)
             messages4.append(message)
+
+            if len(messages4) >= 2:
+                messages4 = messages4[-2:]
+
             response = chat_model.invoke(messages4)
             messages4.append(AIMessage(content=response.content))
 
-            if len(messages4) >= 1:
-                messages4 = messages4[-1:]
-
             response_message = response.content
 
     history[-1][1] = ""
@@ -391,11 +393,12 @@ def File_Interact(history,filepath):
             """
             message = HumanMessage(content=augmented_prompt)
             messages5.append(message)
-            response = chat_model.invoke(messages5)
-            messages5.append(AIMessage(content=response.content))
 
             if len(messages5) >= 1:
                 messages5 = messages5[-1:]
+                            
+            response = chat_model.invoke(messages5)
+            messages5.append(AIMessage(content=response.content))
 
             response_message = response.content