Update app.py

app.py

@@ -206,8 +206,10 @@ AIMessage(content="I'm great thank you. How can I help you?")]
 def Chat_Message(history, messages1):
 
     message=HumanMessage(content=history[-1][0])
+    if isinstance(messages1[-1], HumanMessage):
+        messages1=messages1[:-2]
+        
     messages1.append(message)
-
     if len(messages1) >= 8:
       messages1 = messages1[-8:]
         
@@ -218,8 +220,7 @@ def Chat_Message(history, messages1):
         start_index = error_message.find("Input validation error:")
         end_index = error_message.find("and 4096 `max_new_tokens`")
         if start_index != -1 and end_index != -1:
-            messages1 = messages1[:-2]
-            raise gr.Error(error_message[start_index:end_index+ len("and 4096 `max_new_tokens`")].strip()) from e
+            raise gr.Error(error_message[start_index:end_index].strip()) from e
         else:
             raise gr.Error("Error occurred during response") from e
             
@@ -234,6 +235,8 @@ def Chat_Message(history, messages1):
 def Internet_Search(history, messages2):
 
     message=history[-1][0]
+    if isinstance(messages2[-1], HumanMessage):
+        messages2=messages2[:-2]
 
     similar_docs = db.similarity_search(message, k=3)
 
@@ -250,7 +253,7 @@ def Internet_Search(history, messages2):
     
     #If the query is not related to specific Search Content, engage in general conversation or provide relevant information from other sources.
     """
-
+            
     msg=HumanMessage(content=augmented_prompt)
     messages2.append(msg)
 
@@ -264,8 +267,7 @@ def Internet_Search(history, messages2):
         start_index = error_message.find("Input validation error:")
         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+ len("and 4096 `max_new_tokens`")].strip()) from e
+            raise gr.Error(error_message[start_index:end_index ].strip()) from e
         else:
             raise gr.Error("Error occurred during response") from e
             
@@ -280,6 +282,9 @@ def Internet_Search(history, messages2):
 def Chart_Generator(history, messages3):
 
     message = history[-1][0]
+    if isinstance(messages3[-1], HumanMessage):
+        messages3=messages3[:-2]
+        
     if '#chart' in message:
         message = message.split('#chart', 1)[1].strip()
         chart_url = f"https://quickchart.io/natural/{message}"
@@ -302,8 +307,7 @@ def Chart_Generator(history, messages3):
                 start_index = error_message.find("Input validation error:")
                 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+ len("and 4096 `max_new_tokens`")].strip()) from e
+                    raise gr.Error(error_message[start_index:end_index ].strip()) from e
                 else:
                     raise gr.Error("Error occurred during response") from e
             
@@ -327,8 +331,7 @@ def Chart_Generator(history, messages3):
             start_index = error_message.find("Input validation error:")
             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+ len("and 4096 `max_new_tokens`")].strip()) from e
+                raise gr.Error(error_message[start_index:end_index ].strip()) from e
             else:
                 raise gr.Error("Error occurred during response") from e
         
@@ -345,6 +348,8 @@ def Chart_Generator(history, messages3):
 def Link_Scratch(history, messages4):
 
     combined_message = history[-1][0]
+    if isinstance(messages4[-1], HumanMessage):
+        messages4=messages4[:-2]
 
     link = ""
     user_message = ""
@@ -386,8 +391,7 @@ def Link_Scratch(history, messages4):
                 start_index = error_message.find("Input validation error:")
                 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+ len("and 4096 `max_new_tokens`")].strip()) from e
+                    raise gr.Error(error_message[start_index:end_index ].strip()) from e
                 else:
                     raise gr.Error("Error occurred during response") from e
             
@@ -418,6 +422,8 @@ def display_file_name(file):
 def File_Interact(history, filepath, messages5):
 
     combined_message = history[-1][0]
+    if isinstance(messages5[-1], HumanMessage):
+        messages5=messages5[:-2]
     
     link = ""
     user_message = ""
@@ -455,8 +461,7 @@ def File_Interact(history, filepath, messages5):
                 start_index = error_message.find("Input validation error:")
                 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+ len("and 4096 `max_new_tokens`")].strip()) from e
+                    raise gr.Error(error_message[start_index:end_index ].strip()) from e
                 else:
                     raise gr.Error("Error occurred during response") from e
             
@@ -473,6 +478,8 @@ def File_Interact(history, filepath, messages5):
 def Explore_WebSite(history, messages6):
 
     message=history[-1][0]
+    if isinstance(messages6[-1], HumanMessage):
+        messages6=messages6[:-2]
 
     links = [
         'https://huggingface.co/mou3az'
@@ -504,8 +511,7 @@ def Explore_WebSite(history, messages6):
         start_index = error_message.find("Input validation error:")
         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+ len("and 4096 `max_new_tokens`")].strip()) from e
+            raise gr.Error(error_message[start_index:end_index ].strip()) from e
         else:
             raise gr.Error("Error occurred during response") from e