Update app.py

app.py

@@ -105,6 +105,8 @@ def _launch_demo(args, model, tokenizer):
     )
 
     def predict(_chatbot, task_history):
+        if not _chatbot:
+            return _chatbot 
         chat_query = _chatbot[-1][0]
         query = task_history[-1][0]
         print("User: " + _parse_text(query))
@@ -171,6 +173,9 @@ def _launch_demo(args, model, tokenizer):
         return history, task_history, ""
 
     def add_file(history, task_history, file):
+        if file is None:
+            return history, task_history  # Return if no file is uploaded
+        file_path = file.name
         history = history + [((file.name,), None)]
         task_history = task_history + [((file.name,), None)]
         return history, task_history
@@ -201,7 +206,6 @@ def _launch_demo(args, model, tokenizer):
                     empty_bin = gr.Button("🧹 Clear History")
                 task_history = gr.State([])
 
-        # Linking the buttons and inputs to their functions
         submit_btn.click(
             fn=predict,
             inputs=[chatbot, task_history],