MohammadReza-Halakoo commited on
Commit
810036c
·
verified ·
1 Parent(s): 64a5ea0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +17 -2
app.py CHANGED
@@ -10,13 +10,28 @@ import cv2
10
  from PIL import Image
11
  import pypdfium2
12
  import pytesseract
13
- import streamlit as st
 
 
 
14
 
15
- # ===== Safe runtime dir for Streamlit/HF cache =====
16
  runtime_dir = os.path.join(tempfile.gettempdir(), ".streamlit")
17
  os.environ["STREAMLIT_RUNTIME_DIR"] = runtime_dir
 
18
  os.makedirs(runtime_dir, exist_ok=True)
19
 
 
 
 
 
 
 
 
 
 
 
 
 
20
  # ===== Try to import Surya APIs =====
21
  DET_AVAILABLE = True
22
  try:
 
10
  from PIL import Image
11
  import pypdfium2
12
  import pytesseract
13
+ # --- set safe dirs before importing streamlit ---
14
+ safe_home = os.environ.get("HOME") or "/app"
15
+ os.environ["HOME"] = safe_home
16
+ os.makedirs(os.path.join(safe_home, ".streamlit"), exist_ok=True)
17
 
 
18
  runtime_dir = os.path.join(tempfile.gettempdir(), ".streamlit")
19
  os.environ["STREAMLIT_RUNTIME_DIR"] = runtime_dir
20
+ os.environ["STREAMLIT_CONFIG_DIR"] = os.path.join(safe_home, ".streamlit")
21
  os.makedirs(runtime_dir, exist_ok=True)
22
 
23
+ import streamlit as st
24
+
25
+
26
+
27
+
28
+ import streamlit as st
29
+
30
+ # ===== Safe runtime dir for Streamlit/HF cache =====
31
+ # runtime_dir = os.path.join(tempfile.gettempdir(), ".streamlit")
32
+ # os.environ["STREAMLIT_RUNTIME_DIR"] = runtime_dir
33
+ # os.makedirs(runtime_dir, exist_ok=True)
34
+
35
  # ===== Try to import Surya APIs =====
36
  DET_AVAILABLE = True
37
  try: