update dropdown

app.py

@@ -265,7 +265,7 @@ References
 """
 
 # Function to handle model selection
-def handle_model_selection(model_name):
+def handle_model_selection(model_name, download_status):
     # Inform user that download is starting
     # gr.Info(f"Downloading model weights for {model_name}...")
     print(f"Downloading model weights for {model_name}...")
@@ -273,10 +273,12 @@ def handle_model_selection(model_name):
     model_weights_url = assets["model_weights"]
     download_flag = True
     try:
-        for model_weight in model_weights_url:
+        total_files = len(model_weights_url)
+        for idx, model_weight in enumerate(model_weights_url):
             weight_file = os.path.join(ASSET_DIR, model_weight.split("/")[-1])
             print(weight_file)
             if not os.path.exists(weight_file):
+                download_status = f"Downloading {idx + 1} of {total_files}"
                 download_model(model_weight, weight_file)
             
             if not os.path.exists(weight_file):
@@ -284,12 +286,14 @@ def handle_model_selection(model_name):
                 break
             
         if download_flag:
-            gr.Info(f"Model {model_name} is ready for prediction!")
+            download_status =  f"Model {model_name} is ready for prediction!"
         else:
-            gr.Error(f"An error occurred while downloading model weights!")
+            download_status = f"An error occurred while downloading model weights."
             
     except Exception as e:
-        gr.Error(f"An error occurred while downloading model weights: {e}")
+        download_status = f"An error occurred while downloading model weights."
+        
+    return download_status
 
 
 with gr.Blocks(theme = seafoam, css = css, js = js) as demo:
@@ -301,7 +305,10 @@ with gr.Blocks(theme = seafoam, css = css, js = js) as demo:
     # add dropdown for model selection
     model_names = ['BirdAST', 'BirdAST_Seq', 'EfficientNet']
     model_dropdown = gr.Dropdown(label="Choose a model", choices=model_names)
-    model_dropdown.change(handle_model_selection, inputs=model_dropdown)
+    download_status = gr.Textbox(label="Model Status", lines=3, value='', interactive=False) # Non-interactive textbox for status
+
+    model_dropdown.change(handle_model_selection, inputs=[model_dropdown, download_status], outputs=download_status)
+
     
     with gr.Row():
         with gr.Column(elem_classes="column-container"):