predict.py

import datetime
from genconvit.prediction import genconvit_video_prediction

def predict(video_path: str, query_id: str, factor: int):
    start = datetime.datetime.now()
    try:
        result = genconvit_video_prediction(video_path, factor)
        end = datetime.datetime.now()
        processing_time = end - start
        
        output = {
            'score': result.get('score', 0),
            'frames_processed': result.get('frames_processed', 0),
            'processing_time': str(processing_time),
            'status': 'Success' if result.get('score', 0) >= 0.5 else 'Failed'
        }
        
        return output
        
    except Exception as e:
        return {
            'error': str(e),
            'status': 'Error'
        }