Spaces:
Running
Running
File size: 4,348 Bytes
a236811 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 |
from fastapi import APIRouter, UploadFile, File, Depends, HTTPException
from typing import Dict, Any
from app.services.file_service import file_service
from app.utils.dependencies import get_current_user
from app.models.user import TokenData
router = APIRouter(prefix="/files", tags=["Files"])
@router.post("/upload/image", response_model=Dict[str, Any])
async def upload_image(
file: UploadFile = File(..., description="Image file (JPG, PNG, WEBP)"),
current_user: TokenData = Depends(get_current_user)
):
"""
📷 Upload image with OCR text extraction.
- Extracts text from image using Tesseract OCR
- Saves file to user's folder
- Max size: 10MB
"""
try:
result = await file_service.process_image(file, current_user.user_id)
return {"success": True, "data": result}
except HTTPException:
raise
except Exception as e:
raise HTTPException(500, f"Image upload failed: {str(e)}")
@router.post("/upload/pdf", response_model=Dict[str, Any])
async def upload_pdf(
file: UploadFile = File(..., description="PDF document"),
current_user: TokenData = Depends(get_current_user)
):
"""
📄 Upload PDF with text extraction.
- Extracts all text from PDF pages
- Returns page count
- Max size: 10MB
"""
try:
result = await file_service.process_pdf(file, current_user.user_id)
return {"success": True, "data": result}
except HTTPException:
raise
except Exception as e:
raise HTTPException(500, f"PDF upload failed: {str(e)}")
@router.post("/upload/document", response_model=Dict[str, Any])
async def upload_document(
file: UploadFile = File(..., description="DOCX or TXT file"),
current_user: TokenData = Depends(get_current_user)
):
"""
📝 Upload DOCX or TXT document.
- Extracts text content
- Supports DOCX and TXT formats
- Max size: 10MB
"""
try:
if file.content_type == "application/vnd.openxmlformats-officedocument.wordprocessingml.document":
result = await file_service.process_docx(file, current_user.user_id)
elif file.content_type == "text/plain":
result = await file_service.process_text_file(file, current_user.user_id)
else:
raise HTTPException(400, "Unsupported document type. Use DOCX or TXT.")
return {"success": True, "data": result}
except HTTPException:
raise
except Exception as e:
raise HTTPException(500, f"Document upload failed: {str(e)}")
@router.post("/upload/audio", response_model=Dict[str, Any])
async def upload_audio(
file: UploadFile = File(..., description="Audio file (MP3, WAV, WEBM, OGG, M4A)"),
current_user: TokenData = Depends(get_current_user)
):
"""
🎤 Transcribe audio to text using OpenAI Whisper.
- Supports MP3, WAV, WEBM, OGG, M4A
- Returns full transcription
- Max size: 10MB
- Requires OPENAI_API_KEY in environment
"""
try:
result = await file_service.transcribe_audio(file, current_user.user_id)
return {"success": True, "data": result}
except HTTPException:
raise
except Exception as e:
raise HTTPException(500, f"Audio transcription failed: {str(e)}")
@router.delete("/delete")
async def delete_file(
file_path: str,
current_user: TokenData = Depends(get_current_user)
):
"""
🗑️ Delete uploaded file.
- Requires file_path (relative path from upload dir)
- User can only delete their own files
"""
try:
success = file_service.delete_file(file_path, current_user.user_id)
if not success:
raise HTTPException(404, "File not found or access denied")
return {"success": True, "message": "File deleted successfully"}
except HTTPException:
raise
except Exception as e:
raise HTTPException(500, f"File deletion failed: {str(e)}")
@router.get("/health")
async def file_service_health():
"""🏥 Health check for file service"""
return {
"status": "healthy",
"service": "file_upload",
"supported_formats": {
"images": ["JPG", "PNG", "WEBP"],
"documents": ["PDF", "DOCX", "TXT"],
"audio": ["MP3", "WAV", "WEBM", "OGG", "M4A"]
}
} |