push fix

app.py

@@ -3,12 +3,12 @@ import json
 import os
 import tempfile
 import io
+import warnings
 from pathlib import Path
 from typing import Optional, Tuple
 from fastapi import FastAPI, UploadFile, File, HTTPException
 from fastapi.responses import JSONResponse
 from transformers import AutoTokenizer, AutoModelForCausalLM
-from pdfminer.high_level import extract_text as extract_pdf_text
 from docx import Document as DocxDocument
 from pptx import Presentation
 import logging
@@ -17,9 +17,14 @@ import pytesseract
 from pdf2image import convert_from_path
 import easyocr
 
+warnings.filterwarnings("ignore", category=UserWarning, module="pdfminer")
+warnings.filterwarnings("ignore", category=FutureWarning, module="transformers")
+
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
+from pdfminer.high_level import extract_text as extract_pdf_text
+
 app = FastAPI(
     title="Deckgpt",
     description="Upload your startup pitch deck (PDF, PPT, DOCX) and get an investor-style review",
@@ -49,13 +54,22 @@ async def load_model():
         try:
             ocr_model_dir = os.path.join(tempfile.gettempdir(), '.EasyOCR')
             os.makedirs(ocr_model_dir, exist_ok=True)
+            os.chmod(ocr_model_dir, 0o777)
             
-            ocr_reader = easyocr.Reader(
-                ['en'],
-                gpu=torch.cuda.is_available(),
-                model_storage_directory=ocr_model_dir
-            )
-            logger.info("✅ OCR reader loaded successfully!")
+            old_home = os.environ.get('HOME')
+            try:
+                os.environ['HOME'] = tempfile.gettempdir()
+                ocr_reader = easyocr.Reader(
+                    ['en'],
+                    gpu=torch.cuda.is_available(),
+                    model_storage_directory=ocr_model_dir
+                )
+                logger.info("✅ OCR reader loaded successfully!")
+            finally:
+                if old_home:
+                    os.environ['HOME'] = old_home
+                elif 'HOME' in os.environ:
+                    del os.environ['HOME']
         except Exception as e:
             logger.warning(f"⚠️ EasyOCR failed to load, will use pytesseract fallback: {e}")
             ocr_reader = None