Update app.py

app.py

@@ -216,11 +216,10 @@ def Chat_Message(history, messages1):
     except Exception as e:
         error_message = str(e)
         start_index = error_message.find("Input validation error:")
-        end_index = error_message.find("\n")
+        end_index = error_message.find("and 4096 `max_new_tokens`")
         if start_index != -1 and end_index != -1:
-            messages1.pop()
-            messages1.pop()
-            raise gr.Error(error_message[start_index:end_index].strip()) from e
+            messages1 = messages1[:-2]
+            raise gr.Error(error_message[start_index:end_index+ len("and 4096 `max_new_tokens`")].strip()) from e
         else:
             raise gr.Error("Error occurred during response") from e
             
@@ -266,7 +265,7 @@ def Internet_Search(history, messages2):
         end_index = error_message.find("and 4096 `max_new_tokens`")
         if start_index != -1 and end_index != -1:
             messages2 = messages2[:-2]
-            raise gr.Error(error_message[start_index:end_index].strip()) from e
+            raise gr.Error(error_message[start_index:end_index+ len("and 4096 `max_new_tokens`")].strip()) from e
         else:
             raise gr.Error("Error occurred during response") from e
             
@@ -304,7 +303,7 @@ def Chart_Generator(history, messages3):
                 end_index = error_message.find("and 4096 `max_new_tokens`")
                 if start_index != -1 and end_index != -1:
                     messages3 = messages3[:-2]
-                    raise gr.Error(error_message[start_index:end_index].strip()) from e
+                    raise gr.Error(error_message[start_index:end_index+ len("and 4096 `max_new_tokens`")].strip()) from e
                 else:
                     raise gr.Error("Error occurred during response") from e
             
@@ -329,7 +328,7 @@ def Chart_Generator(history, messages3):
             end_index = error_message.find("and 4096 `max_new_tokens`")
             if start_index != -1 and end_index != -1:
                 messages3 = messages3[:-2]
-                raise gr.Error(error_message[start_index:end_index].strip()) from e
+                raise gr.Error(error_message[start_index:end_index+ len("and 4096 `max_new_tokens`")].strip()) from e
             else:
                 raise gr.Error("Error occurred during response") from e
         
@@ -388,7 +387,7 @@ def Link_Scratch(history, messages4):
                 end_index = error_message.find("and 4096 `max_new_tokens`")
                 if start_index != -1 and end_index != -1:
                     messages4 = messages4[:-2]
-                    raise gr.Error(error_message[start_index:end_index].strip()) from e
+                    raise gr.Error(error_message[start_index:end_index+ len("and 4096 `max_new_tokens`")].strip()) from e
                 else:
                     raise gr.Error("Error occurred during response") from e
             
@@ -457,7 +456,7 @@ def File_Interact(history, filepath, messages5):
                 end_index = error_message.find("and 4096 `max_new_tokens`")
                 if start_index != -1 and end_index != -1:
                     messages5 = messages5[:-2]
-                    raise gr.Error(error_message[start_index:end_index].strip()) from e
+                    raise gr.Error(error_message[start_index:end_index+ len("and 4096 `max_new_tokens`")].strip()) from e
                 else:
                     raise gr.Error("Error occurred during response") from e
             
@@ -506,7 +505,7 @@ def Explore_WebSite(history, messages6):
         end_index = error_message.find("and 4096 `max_new_tokens`")
         if start_index != -1 and end_index != -1:
             messages6 = messages6[:-2]
-            raise gr.Error(error_message[start_index:end_index].strip()) from e
+            raise gr.Error(error_message[start_index:end_index+ len("and 4096 `max_new_tokens`")].strip()) from e
         else:
             raise gr.Error("Error occurred during response") from e