linter fix

Author: DostavalovID

genai/app/api/endpoints/handwritten.py

@@ -1,13 +1,7 @@
 import logging
 from fastapi import APIRouter, File, UploadFile, HTTPException, Depends, Header, Path, status
 from fastapi.responses import FileResponse
-from app.models.schemas import (
-    HandwrittenUploadResponse,
-    HandwrittenListResponse, 
-    HandwrittenStatusResponse,
-    HandwrittenDeleteResponse,
-    HandwrittenDocument
-)
+from app.models.schemas import HandwrittenUploadResponse, HandwrittenListResponse, HandwrittenStatusResponse, HandwrittenDeleteResponse, HandwrittenDocument
 from app.celery_tasks.document_tasks import process_handwritten_document_task
 from app.services.handwritten_storage import get_handwritten_storage, HandwrittenDocumentStorage
 from celery.result import AsyncResult
@@ -27,49 +21,39 @@ async def upload_handwritten_document(
     Upload a handwritten PDF document for processing to LaTeX.
     """
     logger.info(f"Received handwritten document upload: {file.filename} for group {group_id}")
-    
+
     if not file.filename:
         raise HTTPException(status_code=400, detail="No file name provided.")
-    
-    if not file.filename.lower().endswith('.pdf'):
-        raise HTTPException(
-            status_code=400,
-            detail="Only PDF files are supported for handwritten document processing."
-        )
-    
+
+    if not file.filename.lower().endswith(".pdf"):
+        raise HTTPException(status_code=400, detail="Only PDF files are supported for handwritten document processing.")
+
     try:
         # Read file content
         contents = await file.read()
         logger.info(f"Read {len(contents)} bytes from uploaded file: {file.filename}")
-        
+
         if not contents:
             raise HTTPException(status_code=400, detail="Uploaded file content is empty.")
-        
+
         # Start Celery task first to get the real task ID
         task = process_handwritten_document_task.delay(contents, file.filename, group_id, None)
-        
+
         # Use the Celery task ID for storage
         celery_task_id = task.id
-        
+
         # Store metadata using the Celery task ID
         storage.store_document_metadata(celery_task_id, group_id, file.filename)
-        
+
         logger.info(f"Started handwritten processing task {celery_task_id} for {file.filename} in group {group_id}")
-        
-        return HandwrittenUploadResponse(
-            task_id=celery_task_id,
-            filename=file.filename,
-            message="Handwritten document uploaded and processing started in background."
-        )
-        
+
+        return HandwrittenUploadResponse(task_id=celery_task_id, filename=file.filename, message="Handwritten document uploaded and processing started in background.")
+
     except HTTPException as http_exc:
         raise http_exc
     except Exception as e:
         logger.error(f"Unexpected error during handwritten document upload of {file.filename}: {e}", exc_info=True)
-        raise HTTPException(
-            status_code=500,
-            detail=f"An unexpected server error occurred: {str(e)}"
-        )
+        raise HTTPException(status_code=500, detail=f"An unexpected server error occurred: {str(e)}")
     finally:
         await file.close()
 
@@ -83,18 +67,15 @@ def list_handwritten_documents(
     List all handwritten documents for a group, separated by processing status.
     """
     logger.info(f"Listing handwritten documents for group {group_id}")
-    
+
     try:
         documents = storage.get_documents_by_group(group_id)
-        
+
         processing = [doc for doc in documents if doc.status in ["PENDING", "PROCESSING"]]
         completed = [doc for doc in documents if doc.status in ["SUCCESS", "FAILURE"]]
-        
-        return HandwrittenListResponse(
-            processing=processing,
-            completed=completed
-        )
-        
+
+        return HandwrittenListResponse(processing=processing, completed=completed)
+
     except Exception as e:
         logger.error(f"Error listing handwritten documents for group {group_id}: {e}", exc_info=True)
         raise HTTPException(status_code=500, detail="Failed to retrieve handwritten documents.")
@@ -109,16 +90,16 @@ def get_handwritten_status(
     Get the processing status of a handwritten document.
     """
     logger.info(f"Checking status for handwritten document task {task_id}")
-    
+
     try:
         # Get from storage first
         document = storage.get_document(task_id)
         if not document:
             raise HTTPException(status_code=404, detail="Document not found.")
-        
+
         # Check Celery task status
         task_result = AsyncResult(task_id, app=process_handwritten_document_task.app)
-        
+
         # Update status if task is complete
         if task_result.ready():
             if task_result.successful():
@@ -127,21 +108,14 @@ def get_handwritten_status(
                 # Refresh document data
                 document = storage.get_document(task_id)
             elif task_result.failed():
-                error_result = {
-                    "status": "FAILURE",
-                    "error_message": str(task_result.result) if task_result.result else "Unknown error"
-                }
+                error_result = {"status": "FAILURE", "error_message": str(task_result.result) if task_result.result else "Unknown error"}
                 storage.update_document_status(task_id, error_result)
                 document = storage.get_document(task_id)
-        
+
         return HandwrittenStatusResponse(
-            task_id=document.task_id,
-            status=document.status,
-            original_filename=document.original_filename,
-            processed_filename=document.processed_filename,
-            error_message=document.error_message
+            task_id=document.task_id, status=document.status, original_filename=document.original_filename, processed_filename=document.processed_filename, error_message=document.error_message
         )
-        
+
     except HTTPException as http_exc:
         raise http_exc
     except Exception as e:
@@ -159,35 +133,25 @@ def download_processed_document(
     Download the processed LaTeX PDF document.
     """
     logger.info(f"Download request for processed document {task_id} from group {group_id}")
-    
+
     try:
         document = storage.get_document(task_id)
         if not document:
             raise HTTPException(status_code=404, detail="Document not found.")
-        
+
         # Validate group membership
         if document.group_id != group_id:
-            raise HTTPException(
-                status_code=403, 
-                detail="Access denied. You can only download documents from your own group."
-            )
-        
+            raise HTTPException(status_code=403, detail="Access denied. You can only download documents from your own group.")
+
         if document.status != "SUCCESS":
-            raise HTTPException(
-                status_code=400, 
-                detail=f"Document is not ready for download. Status: {document.status}"
-            )
-        
+            raise HTTPException(status_code=400, detail=f"Document is not ready for download. Status: {document.status}")
+
         file_path = storage.get_file_path(task_id, "processed")
         if not file_path or not file_path.exists():
             raise HTTPException(status_code=404, detail="Processed file not found.")
-        
-        return FileResponse(
-            path=str(file_path),
-            media_type="application/pdf",
-            filename=document.processed_filename or f"processed_{document.original_filename}"
-        )
-        
+
+        return FileResponse(path=str(file_path), media_type="application/pdf", filename=document.processed_filename or f"processed_{document.original_filename}")
+
     except HTTPException as http_exc:
         raise http_exc
     except Exception as e:
@@ -204,25 +168,22 @@ def delete_handwritten_document(
     Delete a handwritten document and all its associated files.
     """
     logger.info(f"Delete request for handwritten document {task_id}")
-    
+
     try:
         document = storage.get_document(task_id)
         if not document:
             raise HTTPException(status_code=404, detail="Document not found.")
-        
+
         # Delete the document
         success = storage.delete_document(task_id)
-        
+
         if success:
-            return HandwrittenDeleteResponse(
-                task_id=task_id,
-                message="Document successfully deleted."
-            )
+            return HandwrittenDeleteResponse(task_id=task_id, message="Document successfully deleted.")
         else:
             raise HTTPException(status_code=500, detail="Failed to delete document.")
-            
+
     except HTTPException as http_exc:
         raise http_exc
     except Exception as e:
         logger.error(f"Error deleting document {task_id}: {e}", exc_info=True)
-        raise HTTPException(status_code=500, detail="Failed to delete document.") 
+        raise HTTPException(status_code=500, detail="Failed to delete document.")

genai/app/celery_tasks/document_tasks.py

@@ -34,65 +34,66 @@ def process_and_index_document_task(file_content: bytes, filename: str, tenant:
 def process_handwritten_document_task(file_content: bytes, filename: str, group_id: str, task_id: str = None) -> Dict:
     """
     Celery task to process a handwritten PDF document, converting it to LaTeX PDF.
-    
+
     Args:
         file_content: The PDF file content as bytes
         filename: Original filename
         group_id: Group ID for organization
         task_id: Task ID for tracking (if None, will use Celery task ID)
-    
+
     Returns:
         Dict with processing results including file paths and metadata
     """
     # Use Celery task ID if no custom task_id provided
     if task_id is None:
         task_id = process_handwritten_document_task.request.id
-    
+
     logger.info(f"Starting handwritten document processing for {filename} in group {group_id}, task {task_id}")
-    
+
     # Create directories for handwritten documents
     handwritten_dir = Path("app/data/handwritten")
     handwritten_dir.mkdir(parents=True, exist_ok=True)
-    
+
     group_dir = handwritten_dir / group_id
     group_dir.mkdir(parents=True, exist_ok=True)
-    
+
     task_dir = group_dir / task_id
     task_dir.mkdir(parents=True, exist_ok=True)
-    
+
     try:
         # Save the original PDF file
         original_pdf_path = task_dir / f"original_{filename}"
         with open(original_pdf_path, "wb") as f:
             f.write(file_content)
-        
+
         logger.info(f"Saved original PDF to {original_pdf_path}")
-        
+
         # Process the PDF to LaTeX PDF
         processed_pdf_path = process_pdf_to_latex_pdf(str(original_pdf_path))
-        
+
         if processed_pdf_path and Path(processed_pdf_path).exists():
             # Move the processed file to our managed directory
             processed_filename = f"processed_{Path(filename).stem}.pdf"
             final_processed_path = task_dir / processed_filename
-            
+
             # Move the file to our managed location
             import shutil
+
             shutil.move(processed_pdf_path, final_processed_path)
-            
+
             # Clean up any temporary LaTeX files
-            tex_file = Path(str(original_pdf_path).replace('.pdf', '.tex'))
+            tex_file = Path(str(original_pdf_path).replace(".pdf", ".tex"))
             if tex_file.exists():
                 tex_file.unlink()
-            
+
             # Clean up any other LaTeX compilation artifacts
-            for ext in ['.aux', '.log', '.fls', '.fdb_latexmk']:
-                artifact_file = Path(str(original_pdf_path).replace('.pdf', ext))
+            for ext in [".aux", ".log", ".fls", ".fdb_latexmk"]:
+                artifact_file = Path(str(original_pdf_path).replace(".pdf", ext))
                 if artifact_file.exists():
                     artifact_file.unlink()
-            
+
             logger.info(f"Successfully processed {filename} to {final_processed_path}")
-            
+
             return {
                 "task_id": task_id,
                 "group_id": group_id,
@@ -102,7 +103,7 @@ def process_handwritten_document_task(file_content: bytes, filename: str, group_
                 "processed_path": str(final_processed_path),
                 "status": "SUCCESS",
                 "timestamp": datetime.now().isoformat(),
-                "error_message": None
+                "error_message": None,
             }
         else:
             logger.error(f"Failed to process {filename} - no output file generated")
@@ -115,9 +116,9 @@ def process_handwritten_document_task(file_content: bytes, filename: str, group_
                 "processed_path": None,
                 "status": "FAILURE",
                 "timestamp": datetime.now().isoformat(),
-                "error_message": "Failed to generate processed PDF"
+                "error_message": "Failed to generate processed PDF",
             }
-            
+
     except Exception as e:
         logger.error(f"Error processing handwritten document {filename}: {e}", exc_info=True)
         return {
@@ -129,5 +130,5 @@ def process_handwritten_document_task(file_content: bytes, filename: str, group_
             "processed_path": None,
             "status": "FAILURE",
             "timestamp": datetime.now().isoformat(),
-            "error_message": str(e)
+            "error_message": str(e),
         }