Update app.py

app.py

@@ -25,16 +25,17 @@ import pickle
 # Set up logging
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
 
-# Check if the current directory is writable
-if not os.access(os.getcwd(), os.W_OK):
-    logging.warning("Current directory is not writable. Switching to /tmp.")
-    os.chdir('/tmp')  # Change to a writable directory
+# Set the working directory to a writable location
+WORKING_DIR = "/app"  # Use /app or /tmp
+if not os.path.exists(WORKING_DIR):
+    os.makedirs(WORKING_DIR)
+os.chdir(WORKING_DIR)
 
 # Verify the current directory
 logging.info(f"Current working directory: {os.getcwd()}")
 
 # Cache file to store extracted text
-CACHE_FILE = "/tmp/image_text_cache.pkl"
+CACHE_FILE = os.path.join(WORKING_DIR, "image_text_cache.pkl")
 
 # Load cache if it exists
 if os.path.exists(CACHE_FILE):
@@ -178,7 +179,7 @@ new_images = process_new_images(new_images)
 combined_df.update(new_images)
 
 # Save the updated dataset
-combined_df.to_csv('/tmp/data_with_extracted_text.csv', index=False)
+combined_df.to_csv(os.path.join(WORKING_DIR, "data_with_extracted_text.csv"), index=False)
 
 # Save the cache
 with open(CACHE_FILE, "wb") as f: