updating app.py

app.py

@@ -56,9 +56,16 @@ cfg = write_config()
 def predict(uploaded_file):
     if uploaded_file is None:
         return None, None, "No file uploaded."
-    uploaded_path = uploaded_file
-    input_filename = os.path.basename(uploaded_path)
-    print(f"✅ Image received at {uploaded_path}")
+    # uploaded_path = uploaded_file
+    # input_filename = os.path.basename(uploaded_path)
+    # print(f"✅ Image received at {uploaded_path}")
+
+    uploaded_path = os.path.join(UPLOAD_DIR, uploaded_file.name)
+    with open(uploaded_path, "wb") as f:
+        f.write(uploaded_file.read())
+    print(f"✅ Image saved to {uploaded_path}")
+
+
     # Save uploaded image
     # input_bytes = uploaded_file.read()
     # img = Image.open(BytesIO(input_bytes)).convert("RGB")
@@ -95,8 +102,8 @@ with gr.Blocks() as demo:
 
     with gr.Row():
         with gr.Column():
-            uploaded_file = gr.File(label="Upload your Floorplan Image", type="filepath")
-            # uploaded_file = gr.File(label="Upload your Floorplan Image", type="file")
+            # uploaded_file = gr.File(label="Upload your Floorplan Image", type="filepath")
+            uploaded_file = gr.File(label="Upload your Floorplan Image", type="file")
             run_button = gr.Button("Run Vectorizer 🔥")
 
         with gr.Column():