Update app.py

app.py

@@ -236,17 +236,22 @@ with gr.Blocks(title="MythBuster AI") as iface:
 
         def user_message_handler(message, history, generate_img):
             logger.info(f"User claim: {message}")
+            
+            if history is None:
+                history = []
+        
             response = ask(message)
             history.append({"role": "user", "content": message})
             history.append({"role": "assistant", "content": response})
-
+        
             image_path = None
             if generate_img:
                 funny_prompt = generate_funny_image_prompt(message)
                 image_path = generate_image_from_prompt(funny_prompt, HF_API_TOKEN) 
-
+        
             return "", history, image_path
 
+
     gen_image = gr.Checkbox(label="🎨 Generate Funny Image", value=True)
     submit_btn.click(user_message_handler, [msg, chatbot, gen_image], [msg, chatbot, funny_output])
     msg.submit(user_message_handler, [msg, chatbot, gen_image], [msg, chatbot, funny_output])