Update app.py

app.py

@@ -43,18 +43,22 @@ class FalconChatBot:
     def process_history(self, history):
         if history is None:
             return []
-    
+        
+        # Ensure that history is a list of dictionaries
+        if not isinstance(history, list):
+            return []
+        
         # Filter out special commands from the history
         filtered_history = []
         for message in history:
-            user_message = message["user"]
-            assistant_message = message["assistant"]
-        # Check if the user_message is not a special command
-            if not user_message.startswith("Falcon:"):
-                filtered_history.append({"user": user_message, "assistant": assistant_message})
+            if isinstance(message, dict):
+                user_message = message.get("user", "")
+                assistant_message = message.get("assistant", "")
+                # Check if the user_message is not a special command
+                if not user_message.startswith("Falcon:"):
+                    filtered_history.append({"user": user_message, "assistant": assistant_message})
         return filtered_history
 
-
     def predict(self, system_prompt, user_message, assistant_message, history, temperature=temperature, max_new_tokens=max_new_tokens, top_p=top_p, repetition_penalty=repetition_penalty):
 
         # Process the history to remove special commands