[fix] Return output_dir from build_ui for Hugging Face file access

app.py

@@ -2,8 +2,7 @@ from src.ui import build_ui
 
 if __name__ == "__main__":
     # Build the user interface
-    ui = build_ui()
+    ui, output_dir = build_ui()
     
     # Launch the UI in the browser with access to the "output" folder
-    ui.launch(server_name="0.0.0.0", server_port=7860, allowed_paths=[self.output_dir])
-
+    ui.launch(server_name="0.0.0.0", server_port=7860, allowed_paths=[output_dir])

src/ui.py

@@ -156,4 +156,4 @@ def build_ui(css_path="assets/styles.css"):
                 outputs=[file_download, run_btn, status_message]
             )
 
-    return ui # Return the complete UI
+    return ui, generator.output_dir