Update app.py

app.py

@@ -1,7 +1,3 @@
-# Commented out IPython magic to ensure Python compatibility.
-# %%capture
-# !pip install gradio
-
 import gradio as gr
 from PIL import Image
 import os
@@ -129,7 +125,7 @@ def plan_images_list():
     return [file_ for file_ in os.listdir(images_file) if file_.endswith((".JPG", ".jpg", ".jpeg", ".png", ".PNG"))]
 
 # MAIN FUNCTION ****************************************************************
-def main_function(plan_name, interference = True, aps_int = 0, aps_coor = '(0,0)',
+def main_function(plan_name, uploaded_file, interference = True, aps_int = 0, aps_coor = '(0,0)',
                   apch1  = 0, apch6 = 0, apch11 = 0, coord1 = '(0,0)', coord6 = '(0,0)', coord11 = '(0,0)'):
 
     plan_name = str(plan_name)
@@ -168,11 +164,18 @@ def main_function(plan_name, interference = True, aps_int = 0, aps_coor = '(0,0)
     # **************************************************************************
 
     # Load plan
-    numero = plan_name.split('.')[0]
-    plan_in = np.array(Image.open(f"{path_main}Scennarios init/Scennarios B/{numero}.png")) / 255
+    if plan_name == "Upload your own image" and uploaded_file is not None:
+        plan_image = np.array(Image.open(uploaded_file.name))/255
+        plan_grayscale = plan_image[:, :, 0]
+        plan_in = 1 - plan_grayscale
+
+    elif plan_name == "Upload your own image" and uploaded_file is None:
+        numero = "100"
+        plan_in = np.array(Image.open(f"{path_main}Scennarios init/Scennarios B/{numero}.png")) / 255
 
-    image_plan_path = os.path.join(images_file, plan_name)
-    imagen1 = Image.open(image_plan_path)
+    else:
+        numero = plan_name.split('.')[0]
+        plan_in = np.array(Image.open(f"{path_main}Scennarios init/Scennarios B/{numero}.png")) / 255
 
     # Some variables init
     deep_count = 0
@@ -616,6 +619,7 @@ with gr.Blocks() as demo:
     # Único bloque para el clic de button2
     button2.click(main_function,
                   inputs=[map_dropdown,
+                          upload_image,
                           enable_interference,
                           simple_dropdown,
                           simple_coords,
@@ -623,4 +627,4 @@ with gr.Blocks() as demo:
                           coords_ch1_input, coords_ch6_input, coords_ch11_input],
                   outputs=[image_ch1, image_ch6, image_ch11, image_ap1, image_ap2, image_cover_final, image_cells])
 
-demo.launch(share = True)
+demo.launch()