Update app.py

app.py

@@ -216,8 +216,9 @@ def Chat_Message(history, 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
+        end_index = error_message.find("\nMake sure 'text-generation' task is supported by the model.")
+        if start_index != -1 and end_index != -1:
+            raise gr.Error(error_message[start_index:end_index].strip()) from e
         else:
             raise gr.Error("Error occurred during response") from e
             
@@ -260,8 +261,9 @@ def Internet_Search(history, messages2):
     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
+        end_index = error_message.find("\nMake sure 'text-generation' task is supported by the model.")
+        if start_index != -1 and end_index != -1:
+            raise gr.Error(error_message[start_index:end_index].strip()) from e
         else:
             raise gr.Error("Error occurred during response") from e
             
@@ -296,8 +298,9 @@ def Chart_Generator(history, messages3):
             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
+                end_index = error_message.find("\nMake sure 'text-generation' task is supported by the model.")
+                if start_index != -1 and end_index != -1:
+                    raise gr.Error(error_message[start_index:end_index].strip()) from e
                 else:
                     raise gr.Error("Error occurred during response") from e
             
@@ -319,8 +322,9 @@ def Chart_Generator(history, messages3):
         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
+            end_index = error_message.find("\nMake sure 'text-generation' task is supported by the model.")
+            if start_index != -1 and end_index != -1:
+                raise gr.Error(error_message[start_index:end_index].strip()) from e
             else:
                 raise gr.Error("Error occurred during response") from e
         
@@ -376,8 +380,9 @@ def Link_Scratch(history, messages4):
             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
+                end_index = error_message.find("\nMake sure 'text-generation' task is supported by the model.")
+                if start_index != -1 and end_index != -1:
+                    raise gr.Error(error_message[start_index:end_index].strip()) from e
                 else:
                     raise gr.Error("Error occurred during response") from e
             
@@ -443,11 +448,11 @@ def File_Interact(history, filepath, messages5):
             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
+                end_index = error_message.find("\nMake sure 'text-generation' task is supported by the model.")
+                if start_index != -1 and end_index != -1:
+                    raise gr.Error(error_message[start_index:end_index].strip()) from e
                 else:
                     raise gr.Error("Error occurred during response") from e
-
             
             messages5.append(AIMessage(content=response.content))
 
@@ -491,8 +496,9 @@ def Explore_WebSite(history, messages6):
     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
+        end_index = error_message.find("\nMake sure 'text-generation' task is supported by the model.")
+        if start_index != -1 and end_index != -1:
+            raise gr.Error(error_message[start_index:end_index].strip()) from e
         else:
             raise gr.Error("Error occurred during response") from e