update

Dockerfile

@@ -0,0 +1,66 @@
+# Use an official Python runtime as a parent image
+FROM python:3.8
+
+# Set the working directory in the container
+WORKDIR /app
+
+# Install system dependencies
+RUN apt-get update && apt-get install -y \
+    build-essential \
+    cmake \
+    curl \
+    ca-certificates \
+    gcc \
+    git \
+    git-lfs \
+    wget \
+    libpq-dev \
+    libsndfile1-dev \
+    libgl1 \
+    unzip \
+    libjpeg-dev \
+    libpng-dev \
+    libgomp1 \
+    && rm -rf /var/lib/apt/lists/* \
+    && apt-get clean
+
+WORKDIR /app
+
+# Create .streamlit/config.toml file
+RUN mkdir -p /app/.streamlit && \
+    echo "[server]\nenableXsrfProtection = false\nenableCORS = false" > /app/.streamlit/config.toml
+
+# Clone the repository
+RUN git clone https://github.com/nguyen-brat/text-remove.git /app/text-remove
+
+# Set the working directory to the cloned repo
+WORKDIR /app/text-remove
+
+RUN mkdir -p /app/text-remove/.streamlit && \
+    echo "[server]\nenableXsrfProtection = false\nenableCORS = false" > /app/text-remove/.streamlit/config.toml
+
+# Install Python dependencies
+COPY requirements.txt .
+RUN pip install --no-cache-dir -r requirements.txt
+
+# Set up Craft
+WORKDIR /app/text-remove/CRAFT-pytorch
+RUN wget --no-check-certificate 'https://docs.google.com/uc?export=download&id=1Jk4eGD7crsqCCg9C9VjCLkMN3ze8kutZ' -O "craft_mlt_25k.pth"
+RUN pip install gdown
+RUN gdown 1XSaFwBkOaFOdtk4Ane3DFyJGPRw6v5bO
+
+# Set up lama
+WORKDIR /app/text-remove/lama
+RUN curl -LJO https://huggingface.co/smartywu/big-lama/resolve/main/big-lama.zip
+RUN unzip big-lama.zip
+
+# Set the working directory back to the root of the project
+WORKDIR /app/text-remove
+
+# Make port 7860 available to the world outside this container
+EXPOSE 7860
+
+# Run app.py when the container launches
+ENTRYPOINT ["streamlit", "run"]
+CMD ["app.py", "--server.port=7860"]
+# CMD ["streamlit", "run", "app.py", "--server.port=7860", "--server.address=0.0.0.0"]

README.md

@@ -1,10 +1,11 @@
 ---
-title: Another Docker Remove Text
-emoji: 🏆
-colorFrom: green
+title: Docker Remove Text
+emoji: 👀
+colorFrom: yellow
 colorTo: red
 sdk: docker
 pinned: false
+app_port: 7860
 ---
 
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -0,0 +1,102 @@
+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
+
+st.title("Text Detection App")
+
+uploaded_file = st.file_uploader("Choose an image file", type=["jpg", "jpeg", "png"])
+
+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)
+    with open(input_file_path, "wb") as f:
+        f.write(uploaded_file.getbuffer())
+    
+    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
+            shutil.rmtree(osp(output_path, "mask"), ignore_errors=True)
+            progress_placeholder.empty()
+            status_text.empty()
+
+st.write("Note: The download button will appear after running text detection.")

requirements.txt

@@ -0,0 +1,19 @@
+streamlit==1.24.0
+torch==2.3.1
+torchvision==0.2.1
+opencv-python==4.10.0.82
+scikit-image==0.19.3
+pyyaml
+tqdm
+easydict
+scikit-learn==1.3.2
+pandas==2.0.3
+pytorch-lightning==2.3.0
+kornia==0.7.2
+hydra-core==1.1.0
+omegaconf==2.1.2
+albumentations==0.5.2
+webdataset==0.2.86
+pillow==6.2.2
+streamlit==1.24.0
+pillow==6.2.2