Update app.py

app.py

@@ -18,6 +18,9 @@ def process_metadata(file):
     # Create category mapping dynamically
     category_mapping = {cat: idx for idx, cat in enumerate(sorted(unique_categories))}
 
+    # Create a formatted string of the mapping
+    mapping_str = "\n".join([f"{cat}: {idx}" for cat, idx in category_mapping.items()])
+
     results = []
     # Process each media item
     for item in data["media_items"]:
@@ -43,8 +46,8 @@ def process_metadata(file):
         }
         results.append(json.dumps(output))
 
-    # Return all results joined by newlines
-    return "\n".join(results)
+    # Return both results and mapping
+    return "\n".join(results), mapping_str
 
 
 def save_text(text):
@@ -66,14 +69,17 @@ with gr.Blocks(title="VL-HF Annotations Converter") as iface:
         input_file = gr.File(label="Upload JSON file exported from Visual Layer")
 
     with gr.Row():
-        output_text = gr.Textbox(label="Converted JSONL output", lines=10)
+        output_text = gr.Textbox(label="Converted JSONL output", lines=10, scale=3)
+        mapping_text = gr.Textbox(label="Category Mapping", lines=10)
+
+    with gr.Row():
         download_btn = gr.DownloadButton(
             label="Download JSONL", value="metadata.jsonl", interactive=True
         )
 
     # Connect the components
     output = input_file.change(
-        fn=process_metadata, inputs=[input_file], outputs=[output_text]
+        fn=process_metadata, inputs=[input_file], outputs=[output_text, mapping_text]
     ).then(  # Chain the save operation immediately after processing
         fn=save_text, inputs=[output_text], outputs=[download_btn]
     )