Update app.py

app.py

@@ -207,11 +207,21 @@ def Chat_Message(history, messages1):
 
     message=HumanMessage(content=history[-1][0])
     messages1.append(message)
-    response = chat_model.invoke(messages1)
-    messages1.append(AIMessage(content=response.content))
 
     if len(messages1) >= 8:
       messages1 = messages1[-8:]
+        
+    try:
+        response = chat_model.invoke(messages1)
+    except Exception as e:
+        error_message = str(e)
+        index = error_message.find("Input validation error:")
+        if index != -1:
+            raise gr.Error(error_message[index:].strip()) from e
+        else:
+            raise gr.Error("Error occurred during response") from e
+            
+    messages1.append(AIMessage(content=response.content))
 
     history[-1][1] = ""
     for character in response.content:
@@ -245,7 +255,16 @@ def Internet_Search(history, messages2):
     if len(messages2) >= 4:
         messages2 = messages2[-4:]
         
-    response = chat_model.invoke(messages2)
+    try:
+        response = chat_model.invoke(messages1)
+    except Exception as e:
+        error_message = str(e)
+        index = error_message.find("Input validation error:")
+        if index != -1:
+            raise gr.Error(error_message[index:].strip()) from e
+        else:
+            raise gr.Error("Error occurred during response") from e
+            
     messages2.append(AIMessage(content=response.content))
 
     history[-1][1] = ""
@@ -272,7 +291,16 @@ def Chart_Generator(history, messages3):
             if len(messages3) >= 6:
                 messages3 = messages3[-6:]
     
-            response = chat_model.invoke(messages3)
+            try:
+                response = chat_model.invoke(messages1)
+            except Exception as e:
+                error_message = str(e)
+                index = error_message.find("Input validation error:")
+                if index != -1:
+                    raise gr.Error(error_message[index:].strip()) from e
+                else:
+                    raise gr.Error("Error occurred during response") from e
+            
             messages3.append(AIMessage(content=response.content))
     
             combined_content = f'{image_html}<br>{response.content}'
@@ -286,7 +314,16 @@ def Chart_Generator(history, messages3):
         if len(messages3) >= 6:
             messages3 = messages3[-6:]
     
-        response = chat_model.invoke(messages3)
+        try:
+            response = chat_model.invoke(messages1)
+        except Exception as e:
+            error_message = str(e)
+            index = error_message.find("Input validation error:")
+            if index != -1:
+                raise gr.Error(error_message[index:].strip()) from e
+            else:
+                raise gr.Error("Error occurred during response") from e
+        
         messages3.append(AIMessage(content=response.content))
 
         combined_content=response.content
@@ -334,7 +371,16 @@ def Link_Scratch(history, messages4):
             if len(messages4) >= 2:
                 messages4 = messages4[-2:]
 
-            response = chat_model.invoke(messages4)
+            try:
+                response = chat_model.invoke(messages1)
+            except Exception as e:
+                error_message = str(e)
+                index = error_message.find("Input validation error:")
+                if index != -1:
+                    raise gr.Error(error_message[index:].strip()) from e
+                else:
+                    raise gr.Error("Error occurred during response") from e
+            
             messages4.append(AIMessage(content=response.content))
 
             response_message = response.content
@@ -392,7 +438,17 @@ def File_Interact(history, filepath, messages5):
             if len(messages5) >= 1:
                 messages5 = messages5[-1:]
                             
-            response = chat_model.invoke(messages5)
+            try:
+                response = chat_model.invoke(messages1)
+            except Exception as e:
+                error_message = str(e)
+                index = error_message.find("Input validation error:")
+                if index != -1:
+                    raise gr.Error(error_message[index:].strip()) from e
+                else:
+                    raise gr.Error("Error occurred during response") from e
+
+            
             messages5.append(AIMessage(content=response.content))
 
             response_message = response.content
@@ -430,7 +486,16 @@ def Explore_WebSite(history, messages6):
     if len(messages6) >= 2:
         messages6 = messages6[-2:]
         
-    response = chat_model.invoke(messages6)
+    try:
+        response = chat_model.invoke(messages1)
+    except Exception as e:
+        error_message = str(e)
+        index = error_message.find("Input validation error:")
+        if index != -1:
+            raise gr.Error(error_message[index:].strip()) from e
+        else:
+            raise gr.Error("Error occurred during response") from e
+    
     messages6.append(AIMessage(content=response.content))
 
     history[-1][1] = ""