import requests
from django.http import JsonResponse
from .models import VideoGeneration
from django.views.decorators.csrf import csrf_exempt
from django.utils.decorators import method_decorator

@method_decorator(csrf_exempt, name='dispatch')
def text_to_video(request):
    """
    接口：根据文本描述生成视频
    """
    if request.method == "POST":
        try:
            user = request.user
            data = request.POST

            # 生成视频ID
            video_id = generate_video_id(user.id)

            # 获取参数
            text_prompt = data.get('text_prompt')
            model = data.get('model', 'gen3')
            width = int(data.get('width', 640))
            height = int(data.get('height', 480))
            motion = int(data.get('motion', 5))
            seed = int(data.get('seed', 0))
            upscale = data.get('upscale', 'false') == 'true'
            interpolate = data.get('interpolate', 'false') == 'true'
            callback_url = data.get('callback_url', '')
            time_duration = int(data.get('time', 5))

            # 准备POST请求数据
            payload = {
                "text_prompt": text_prompt,
                "model": model,
                "width": width,
                "height": height,
                "motion": motion,
                "seed": seed,
                "upscale": upscale,
                "interpolate": interpolate,
                "callback_url": callback_url,
                "time": time_duration
            }

            # 发送请求到API
            response = requests.post('https://api.aivideoapi.com/runway/generate/text', json=payload)

            # 解析响应数据
            if response.status_code == 200:
                res_data = response.json()
                uuid = res_data.get('uuid')

                # 保存生成记录到数据库
                video_generation = VideoGeneration.objects.create(
                    video_id=video_id,
                    user=user,
                    text=text_prompt,
                    status='in_progress',
                    pid=uuid,
                    ratio=f"{width}:{height}"
                )
                video_generation.save()

                return JsonResponse({'code': 200, 'message': '请求成功', 'data': {'pid': uuid}})

            else:
                return JsonResponse({'code': response.status_code, 'message': 'API请求失败', 'data': {}})

        except Exception as e:
            return JsonResponse({'code': 500, 'message': f'内部错误: {str(e)}', 'data': {}})

    return JsonResponse({'code': 405, 'message': '请求方法错误', 'data': {}})


@method_decorator(csrf_exempt, name='dispatch')
def image_to_video(request):
    """
    接口：根据图片生成视频
    """
    if request.method == "POST":
        try:
            user = request.user
            data = request.POST

            # 生成视频ID
            video_id = generate_video_id(user.id)

            # 获取参数
            text_prompt = data.get('text_prompt')
            model = data.get('model', 'gen3')
            img_prompt = data.get('img_prompt')
            image_as_end_frame = data.get('image_as_end_frame', 'false') == 'true'
            motion = int(data.get('motion', 5))
            seed = int(data.get('seed', 0))
            upscale = data.get('upscale', 'false') == 'true'
            interpolate = data.get('interpolate', 'false') == 'true'
            callback_url = data.get('callback_url', '')
            time_duration = int(data.get('time', 5))

            # 准备POST请求数据
            payload = {
                "text_prompt": text_prompt,
                "model": model,
                "img_prompt": img_prompt,
                "image_as_end_frame": image_as_end_frame,
                "motion": motion,
                "seed": seed,
                "upscale": upscale,
                "interpolate": interpolate,
                "callback_url": callback_url,
                "time": time_duration
            }

            # 发送请求到API
            response = requests.post('https://api.aivideoapi.com/runway/generate/imageDescription', json=payload)

            # 解析响应数据
            if response.status_code == 200:
                res_data = response.json()
                uuid = res_data.get('uuid')

                # 保存生成记录到数据库
                video_generation = VideoGeneration.objects.create(
                    video_id=video_id,
                    user=user,
                    text=text_prompt,
                    status='in_progress',
                    pid=uuid,
                    ratio="auto"
                )
                video_generation.save()

                return JsonResponse({'code': 200, 'message': '请求成功', 'data': {'pid': uuid}})

            else:
                return JsonResponse({'code': response.status_code, 'message': 'API请求失败', 'data': {}})

        except Exception as e:
            return JsonResponse({'code': 500, 'message': f'内部错误: {str(e)}', 'data': {}})

    return JsonResponse({'code': 405, 'message': '请求方法错误', 'data': {}})