Update app.py

app.py

@@ -21,8 +21,15 @@ def save_database(database):
 def format_prompt(message, system, history):
     # Format prompt according to the new template
     prompt = f"SYSTEM: {system}\n<|endofsystem|>\n"
-    for user_prompt, bot_response in history:
-        prompt += f"USER: {user_prompt}\n\n\nASSISTANT: {bot_response}<|endoftext|>\n"
+    
+    for entry in history:
+        if len(entry) == 2:  # Check if the history contains exactly two items
+            user_prompt, bot_response = entry
+            prompt += f"USER: {user_prompt}\n\n\nASSISTANT: {bot_response}<|endoftext|>\n"
+        else:
+            # If the history doesn't match the expected format, handle it gracefully
+            continue
+    
     prompt += f"USER: {message}\n\n\nASSISTANT:"
     return prompt
 
@@ -105,4 +112,4 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
         additional_inputs=additional_inputs,
     )
 
-demo.queue().launch(debug=True)
+demo.queue().launch(debug=True)