Update app.py

app.py

@@ -105,16 +105,12 @@ def _parse_text(text):
     text = "".join(lines)
     return text
     
-def save_image(image_file, uploaded_file_dir):
-    upload_dir = Path(uploaded_file_dir) / "uploads"
-    upload_dir.mkdir(parents=True, exist_ok=True)
+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 = upload_dir / filename
+    file_path = Path(upload_dir) / filename
     with open(file_path, "wb") as f:
-        if isinstance(image_file, gr.NamedString):
-            f.write(image_file.value.encode()) 
-        else:
-            f.write(image_file.read())
+        f.write(image_file.read())
     return str(file_path)
     
 def add_file(history, task_history, file):