update

app.py

@@ -107,6 +107,8 @@ if uploaded_file is not None:
     image = Image.open(uploaded_file)
     # image.save(os.path.join(PROJECT_ROOT, input_file_path))
     image.save(input_file_path)
+    file_name = " || ".join(os.listdir(f'{temp_dir}/target_folder'))
+    st.write(file_name)
     
     if st.button("Run Text Detection"):
         progress_placeholder = st.empty()
@@ -118,7 +120,12 @@ if uploaded_file is not None:
             #os.makedirs(osp(output_path, "result"), exist_ok=True)
             #os.makedirs(osp(output_path, "mask"), exist_ok=True)
             rc, stderr_output = run_bash_script(input_path, output_path, progress_placeholder, status_text)
-            
+            file_namea = " || ".join(os.listdir(f'{temp_dir}/target_folder/mask'))
+            file_nameb = " || ".join(os.listdir(f'{temp_dir}/target_folder/bbox'))
+            file_namec = " || ".join(os.listdir(f'{temp_dir}/target_folder/result'))
+            st.write(file_namea)
+            st.write(file_nameb)
+            st.write(file_namec)
             if rc == 0:
                 status_text.text("Text detection completed successfully!")
                 result_folder = os.path.join(output_path, "result")
@@ -148,10 +155,10 @@ if uploaded_file is not None:
             progress_placeholder.empty()
             status_text.empty()
 
-file_namea = " || ".join(os.listdir(f'{temp_dir}/mask'))
-file_nameb = " || ".join(os.listdir(f'{temp_dir}/result'))
-file_namec = " || ".join(os.listdir(f'{temp_dir}/bbox'))
-st.write(file_namea)
-st.write(file_nameb)
-st.write(file_namec)
+    file_namea = " || ".join(os.listdir(f'{temp_dir}/mask'))
+    file_nameb = " || ".join(os.listdir(f'{temp_dir}/result'))
+    file_namec = " || ".join(os.listdir(f'{temp_dir}/bbox'))
+    st.write(file_namea)
+    st.write(file_nameb)
+    st.write(file_namec)
 st.write("Note: The download button will appear after running text detection.")