Update app.py

app.py

@@ -239,6 +239,15 @@ def verify_api_key(api_key: str = Header(...)):
 def read_root():
     return {"message": "Welcome to the Invoice Summarization API!"}
 
+def get_content_type_from_s3(file_key):
+    """Fetch the content type (MIME type) of a file stored in S3."""
+    try:
+        response = s3_client.head_object(Bucket=S3_BUCKET_NAME, Key=file_key)
+        return response.get('ContentType', 'application/octet-stream')  # Default to binary if not found
+    except Exception as e:
+        raise Exception(f"Failed to get content type from S3: {str(e)}")
+
+
 @app.get("/ocr/extraction")
 def extract_text_from_file(
     api_key: str = Depends(verify_api_key),