Update app.py

app.py

@@ -30,7 +30,6 @@ def fetch_metadata(api_key, workspace, project, version):
     resp = requests.get(endpoint, params=params)
     resp.raise_for_status()
     data = resp.json()
-    # Use version-level image count and project-level class counts
     total_images = data.get('version', {}).get('images') or data.get('project', {}).get('images', 0)
     classes = data.get('project', {}).get('classes', {})
     return total_images, classes
@@ -64,18 +63,25 @@ def make_bar_chart(counts):
     return fig
 
 
-def load_datasets(api_key, file_obj):
-    content = file_obj.read().decode('utf-8')
-    urls = [line.strip() for line in content.splitlines() if line.strip()]
-    total_images, aggregated = aggregate_datasets(api_key, urls)
-    # Prepare dataframe data
+def load_datasets(api_key, file_objs):
+    """
+    Read multiple .txt uploads, parse URLs, and aggregate metadata.
+    Returns total images, dataframe data, plot, and JSON mapping.
+    """
+    # Combine URLs from all uploaded files
+    urls = []
+    for file_obj in file_objs:
+        content = file_obj.read().decode('utf-8')
+        urls.extend([line.strip() for line in content.splitlines() if line.strip()])
+    # Remove duplicate URLs while preserving order
+    unique_urls = list(dict.fromkeys(urls))
+    total_images, aggregated = aggregate_datasets(api_key, unique_urls)
     df_data = [[cls, cnt] for cls, cnt in aggregated.items()]
     fig = make_bar_chart(aggregated)
     return f"{total_images}", df_data, fig, json.dumps(aggregated, indent=2)
 
 
 def update_classes(df_data):
-    # df_data: list of [class_name, count]
     updated = {row[0]: int(row[1]) for row in df_data if row[0] and isinstance(row[1], (int, float, str)) and str(row[1]).isdigit()}
     total_images = sum(updated.values())
     fig = make_bar_chart(updated)
@@ -87,7 +93,7 @@ def build_ui():
         gr.Markdown("## Roboflow Dataset Inspector")
         with gr.Row():
             api_input = gr.Textbox(label="Roboflow API Key", type="password", placeholder="Enter your API key...")
-            file_input = gr.File(label="Upload .txt of Roboflow URLs", file_types=[".txt"] )
+            file_inputs = gr.Files(label="Upload one or more .txt files of Roboflow URLs", file_types=[".txt"])  
         load_btn = gr.Button("Load Datasets")
         total_out = gr.Textbox(label="Total Images", interactive=False)
         df = gr.Dataframe(headers=["Class Name", "Count"], row_count=(1, None), col_count=2, interactive=True)
@@ -96,7 +102,7 @@ def build_ui():
         update_btn = gr.Button("Apply Class Edits")
 
         load_btn.click(fn=load_datasets,
-                       inputs=[api_input, file_input],
+                       inputs=[api_input, file_inputs],
                        outputs=[total_out, df, plot_out, json_out])
 
         update_btn.click(fn=update_classes,