|
|
| from flask import request |
| import numpy as np |
| import traceback |
|
|
| from core.response_json import error_response, success_response, server_error_response |
| from middlewares.auth import token_optional |
| from services.analysis_service import run_deepfake_analysis, save_scan_history |
|
|
|
|
| @token_optional |
| def scan_image(current_user): |
| if 'file' not in request.files: |
| return error_response(message="No file uploaded", status_code=400) |
|
|
| file = request.files['file'] |
|
|
| if file.filename == '': |
| return error_response(message="No selected file", status_code=400) |
|
|
| try: |
| file_bytes = np.frombuffer(file.read(), np.uint8) |
|
|
| analysis_result = run_deepfake_analysis(file_bytes) |
|
|
| if current_user is not None: |
| save_scan_history( |
| user_id=current_user.get("user_id"), |
| file_name=file.filename, |
| result=analysis_result["result"], |
| confidence_score=analysis_result["probability"], |
| processing_time=analysis_result["processing_time"] |
| ) |
| analysis_result["saved_to_history"] = True |
| else: |
| analysis_result["saved_to_history"] = False |
|
|
| return success_response( |
| message="Analysis complete", |
| data=analysis_result |
| ) |
|
|
| except ValueError as ve: |
| return error_response(message=str(ve), status_code=400) |
| except Exception as e: |
| print("ERROR:", str(e)) |
| traceback.print_exc() |
| return server_error_response(error_details=str(e)) |
