Update app.py

app.py

@@ -103,10 +103,25 @@ def _parse_text(text):
                 lines[i] = "<br>" + line
     text = "".join(lines)
     return text
+def save_image(image_file):
+    # Create a directory for uploaded images if it doesn't exist
+    upload_dir = Path(uploaded_file_dir) / "uploads"
+    upload_dir.mkdir(parents=True, exist_ok=True)
 
+    # Generate a unique filename for the image
+    filename = secrets.token_hex(10) + Path(image_file.name).suffix
+
+    # Save the image to the filesystem
+    file_path = upload_dir / filename
+    with open(file_path, "wb") as f:
+        f.write(image_file.read())
+
+    # Return the path to the saved image
+    return str(file_path)
+    
 def add_file(history, task_history, file):
     if file is None:
-        return history, task_history  # Return if no file is uploaded
+        return history, task_history
     file_path = save_image(file)
     history = history + [((file_path,), None)]
     task_history = task_history + [((file_path,), None)]