update

Dockerfile

@@ -69,6 +69,8 @@ RUN chmod 777 $HOME/app/text-remove/test_folder
 # Make port 7860 available to the world outside this container
 EXPOSE 7860
 
+ENV PYTHONPATH=$HOME/app/text-remove
+
 # Run app.py when the container launches
 ENTRYPOINT ["streamlit", "run"]
 CMD ["app.py", "--server.port=7860", "--server.headless=true", "--server.enableCORS=false", "--server.enableXsrfProtection=false", "--server.fileWatcherType=none"]

app.py

@@ -6,8 +6,26 @@ import zipfile
 import io
 import shutil
 import time
+import sys
 from PIL import Image
 
+def GET_PROJECT_ROOT():
+    count = 0
+    # goto the root folder of LogBar
+    current_abspath = os.path.abspath(__file__)
+    while True:
+        if count > 1000:
+            print("Can find root error")
+            sys.exit()
+        if os.path.split(current_abspath)[1] == 'text-remove':
+            project_root = current_abspath
+            break
+        else:
+            current_abspath = os.path.dirname(current_abspath)
+    return project_root
+
+PROJECT_ROOT = GET_PROJECT_ROOT()
+
 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)
@@ -69,7 +87,7 @@ if uploaded_file is not None:
 
     input_file_path = os.path.join(input_path, uploaded_file.name)
     image = Image.open(uploaded_file)
-    image.save(input_file_path)
+    image.save(os.path.join(PROJECT_ROOT, input_file_path))
     
     if st.button("Run Text Detection"):
         progress_placeholder = st.empty()