Update main.py

main.py

@@ -13,14 +13,14 @@ import os
 # Initialize FastAPI app
 app = FastAPI()
 
-# Set the correct path for static files
-STATIC_DIR = r"C:\Users\User\doc_translation_service\translation\static"
+# Set the correct path for static files (for Hugging Face Spaces)
+STATIC_DIR = "static"
 
 # Ensure the static directory exists
 if not os.path.exists(STATIC_DIR):
     os.makedirs(STATIC_DIR)
 
-app.mount("/", StaticFiles(directory="static", html=True), name="static")
+app.mount("/static", StaticFiles(directory=STATIC_DIR), name="static")
 
 @app.get("/", response_class=HTMLResponse)
 async def read_root():
@@ -56,13 +56,11 @@ def load_translator(src_code: str, tgt_code: str):
     
     if model_key in AVAILABLE_MODELS:
         return pipeline("translation", model=AVAILABLE_MODELS[model_key])
-
     elif src_code != "en" and tgt_code != "en":
         return (
             pipeline("translation", model=AVAILABLE_MODELS.get(f"{src_code}-en")),
             pipeline("translation", model=AVAILABLE_MODELS.get(f"en-{tgt_code}"))
         )
-
     else:
         raise ValueError(f"No model available for {src_code} -> {tgt_code}")
 
@@ -75,15 +73,12 @@ def extract_text(file: UploadFile):
     try:
         if file.filename.endswith(".txt"):
             return file.file.read().decode("utf-8")
-
         elif file.filename.endswith(".pdf"):
             doc = fitz.open(stream=file.file.read(), filetype="pdf")
             return "\n".join([page.get_text() for page in doc])
-
         elif file.filename.endswith(".docx"):
             doc = Document(file.file)
             return "\n".join([para.text for para in doc.paragraphs])
-
         elif file.filename.endswith(".xlsx"):
             wb = openpyxl.load_workbook(file.file)
             text = ""
@@ -92,7 +87,6 @@ def extract_text(file: UploadFile):
                 for row in ws.iter_rows():
                     text += "\t".join([str(cell.value or "") for cell in row]) + "\n"
             return text
-
         elif file.filename.endswith(".pptx"):
             prs = Presentation(file.file)
             text = ""
@@ -101,10 +95,8 @@ def extract_text(file: UploadFile):
                     if hasattr(shape, "text"):
                         text += shape.text + "\n"
             return text
-
         else:
             raise HTTPException(status_code=400, detail="File type not supported.")
-
     except Exception as e:
         raise HTTPException(status_code=500, detail=f"Error extracting text: {str(e)}")
 
@@ -126,19 +118,13 @@ async def upload_file(
         raise HTTPException(status_code=400, detail=f"Unsupported language: {src_lang} -> {tgt_lang}")
 
     try:
-        # Load translation model
         translator = load_translator(src_code, tgt_code)
-
-        # If indirect translation via English
         if isinstance(translator, tuple):
             translator1, translator2 = translator
             intermediate_text = "\n".join([translator1(chunk)[0]['translation_text'] for chunk in chunk_text(text)])
             translated_text = "\n".join([translator2(chunk)[0]['translation_text'] for chunk in chunk_text(intermediate_text)])
-
         else:
             translated_text = "\n".join([translator(chunk)[0]['translation_text'] for chunk in chunk_text(text)])
-
         return {"translated_text": translated_text}
-
     except Exception as e:
         raise HTTPException(status_code=500, detail=f"Internal error: {str(e)}")