update

app.py

@@ -0,0 +1,118 @@
+import streamlit as st
+import os
+from os.path import join as osp
+import subprocess
+import zipfile
+import io
+import shutil
+import time
+
+def run_bash_script(input_image_path, output_path, progress_placeholder, status_text):
+    bash_command = f"bash config/text_detection.sh -s {input_image_path} -t {output_path}"
+    process = subprocess.Popen(bash_command, shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True)
+    
+    progress = 0
+    for line in process.stdout:
+        st.text(line.strip())
+        progress += 0.1
+        progress_placeholder.progress(min(progress, 1.0))
+    
+    # Capture and display stderr
+    stderr_output = process.stderr.read()
+    if stderr_output:
+        status_text.error("Error output:")
+        st.code(stderr_output, language="bash")
+    
+    rc = process.wait()
+    return rc, stderr_output
+
+def zip_result_files(result_folder):
+    zip_buffer = io.BytesIO()
+    
+    with zipfile.ZipFile(zip_buffer, "w", zipfile.ZIP_DEFLATED) as zip_file:
+        for root, _, files in os.walk(result_folder):
+            for file in files:
+                if file.endswith(".png"):
+                    file_path = os.path.join(root, file)
+                    arcname = os.path.relpath(file_path, result_folder)
+                    zip_file.write(file_path, arcname)
+    
+    return zip_buffer
+
+def clear_folder(folder_path):
+    for filename in os.listdir(folder_path):
+        file_path = os.path.join(folder_path, filename)
+        try:
+            if os.path.isfile(file_path) or os.path.islink(file_path):
+                os.unlink(file_path)  # Remove file or symlink
+            elif os.path.isdir(file_path):
+                shutil.rmtree(file_path, ignore_errors=True)  # Remove directory and its contents
+        except Exception as e:
+            print(f'Failed to delete {file_path}. Reason: {e}')
+
+st.title("Text Detection App")
+
+uploaded_file = st.file_uploader("Choose an image file", type=["jpg", "jpeg", "png"])
+
+if uploaded_file is not None:
+    input_path = "test_folder"
+    output_path = "target_test"
+    input_file_path = os.path.join(input_path, uploaded_file.name)
+    with open(input_file_path, "wb") as f:
+        f.write(uploaded_file.getbuffer())
+
+if uploaded_file is not None:
+    st.image(uploaded_file, caption="Uploaded Image", use_column_width=True)
+    
+    # Create a temporary directory for processing
+    
+    # Save the uploaded file temporarily
+    input_path = "test_folder"
+    output_path = "target_test"
+    os.makedirs(input_path, exist_ok=True)
+    os.makedirs(osp(output_path, "result"), exist_ok=True)
+    os.makedirs(osp(output_path, "mask"), exist_ok=True)
+
+    input_file_path = os.path.join(input_path, uploaded_file.name)
+    
+    if st.button("Run Text Detection"):
+        progress_placeholder = st.empty()
+        status_text = st.empty()
+        
+        try:
+            status_text.text("Running text detection...")
+            os.makedirs(input_path, exist_ok=True)
+            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)
+            
+            if rc == 0:
+                status_text.text("Text detection completed successfully!")
+                result_folder = os.path.join(output_path, "result")
+                if os.path.exists(result_folder):
+                    st.write("You can now download the results.")
+                    
+                    # Add download button
+                    zip_buffer = zip_result_files(result_folder)
+                    st.download_button(
+                        label="Download Results",
+                        data=zip_buffer.getvalue(),
+                        file_name="text_detection_results.zip",
+                        mime="application/zip"
+                    )
+                else:
+                    st.error("Result folder not found. The text detection might have failed.")
+            else:
+                st.error(f"Text detection failed with return code {rc}")
+                if stderr_output:
+                    st.error("Error details:")
+                    st.code(stderr_output, language="bash")
+        except Exception as e:
+            st.error(f"An error occurred: {str(e)}")
+        finally:
+            # Clean up temporary files
+            clear_folder(osp(output_path, "mask"))
+            progress_placeholder.empty()
+            status_text.empty()
+
+st.write("Note: The download button will appear after running text detection.")