Update app.py

app.py

@@ -42,7 +42,7 @@ def handle_image_submission(_chatbot, task_history, file):
     if file is None:
         print("No file uploaded")
         return _chatbot, task_history  
-    file_path = save_image(file, uploaded_file_dir)  # Renamed for clarity
+    file_path = save_image(file, uploaded_file_dir) 
     history_item = ((file_path,), None)
     _chatbot.append(history_item)
     task_history.append(history_item)
@@ -104,14 +104,18 @@ def _parse_text(text):
                 lines[i] = "<br>" + line
     text = "".join(lines)
     return text
-    
+
 def save_image(image_file, upload_dir):
     Path(upload_dir).mkdir(parents=True, exist_ok=True)
     filename = secrets.token_hex(10) + Path(image_file.name).suffix
     file_path = Path(upload_dir) / filename
-    with open(file_path, "wb") as f:
-        f.write(image_file.read())
+    if isinstance(image_file, gr.NamedString):
+        shutil.copy(image_file.name, file_path)
+    else:
+        with open(file_path, "wb") as f:
+            f.write(image_file.read())
     return str(file_path)
+
     
 def add_file(history, task_history, file):
     if file is None: