Update app.py

app.py

@@ -209,13 +209,13 @@ def Chat_Message(history):
     message=HumanMessage(content=history[-1][0])
     messages1.append(message)
     response = chat_model.invoke(messages1)
-    messages1.append(AIMessage(content=response))
+    messages1.append(AIMessage(content=response.content))
 
     if len(messages1) >= 8:
       messages1 = messages1[-8:]
 
     history[-1][1] = ""
-    for character in response:
+    for character in response.content:
         history[-1][1] += character
         time.sleep(0.0025)
         yield history
@@ -244,13 +244,13 @@ def Web_Search(history):
     msg=HumanMessage(content=augmented_prompt)
     messages2.append(msg)
     response = chat_model.invoke(messages2)
-    messages2.append(AIMessage(content=response))
+    messages2.append(AIMessage(content=response.content))
 
     if len(messages2) >= 8:
       messages2 = messages2[-8:]
 
     history[-1][1] = ""
-    for character in response:
+    for character in response.content:
         history[-1][1] += character
         time.sleep(0.0025)
         yield history
@@ -272,12 +272,12 @@ def Chart_Generator(history):
             messages3.append(prompt)
     
             response = chat_model.invoke(messages3)
-            messages3.append(AIMessage(content=response))
+            messages3.append(AIMessage(content=response.content))
     
             if len(messages3) >= 8:
                 messages3 = messages3[-8:]
     
-            combined_content = f'{image_html}<br>{response}'
+            combined_content = f'{image_html}<br>{response.content}'
         else:
             response_text = "Can't generate this image. Please provide valid chart details."
             combined_content = response_text
@@ -286,12 +286,12 @@ def Chart_Generator(history):
         messages3.append(prompt)
     
         response = chat_model.invoke(messages3)
-        messages3.append(AIMessage(content=response))
+        messages3.append(AIMessage(content=response.content))
     
         if len(messages3) >= 8:
             messages3 = messages3[-8:]
 
-        combined_content=response
+        combined_content=response.content
         
     history[-1][1] = ""
     for character in combined_content:
@@ -334,12 +334,12 @@ def Link_Scratch(history):
             message = HumanMessage(content=augmented_prompt)
             messages4.append(message)
             response = chat_model.invoke(messages4)
-            messages4.append(AIMessage(content=response))
+            messages4.append(AIMessage(content=response.content))
 
             if len(messages4) >= 1:
                 messages4 = messages4[-1:]
 
-            response_message = response
+            response_message = response.content
 
     history[-1][1] = ""
     for character in response_message:
@@ -392,12 +392,12 @@ def File_Interact(history,filepath):
             message = HumanMessage(content=augmented_prompt)
             messages5.append(message)
             response = chat_model.invoke(messages5)
-            messages5.append(AIMessage(content=response))
+            messages5.append(AIMessage(content=response.content))
 
             if len(messages5) >= 1:
                 messages5 = messages5[-1:]
 
-            response_message = response
+            response_message = response.content
 
     history[-1][1] = ""
     for character in response_message: