Add script to create videos for speaking questions.

helper/heygen_api.py

@@ -0,0 +1,87 @@
+import os
+import requests
+import time
+
+from dotenv import load_dotenv
+
+load_dotenv()
+
+# Get HeyGen token
+TOKEN = os.getenv("HEY_GEN_TOKEN")
+
+# POST TO CREATE VIDEO
+CREATE_VIDEO_URL = 'https://api.heygen.com/v1/video.generate'
+GET_VIDEO_URL = 'https://api.heygen.com/v1/video_status.get'
+POST_HEADER = {
+    'X-Api-Key': TOKEN,
+    'Content-Type': 'application/json'
+}
+GET_HEADER = {
+    'X-Api-Key': TOKEN
+}
+
+def create_video(text):
+    # POST TO CREATE VIDEO
+    data = {
+        "background": "#ffffff",
+        "clips": [
+            {
+                "avatar_id": "Mido-lite-20221128",
+                "avatar_style": "normal",
+                "input_text": text,
+                "offset": {
+                    "x": 0,
+                    "y": 0
+                },
+                "scale": 1,
+                "voice_id": "ccb30e87c6b34ca8941f88352c71612d"
+            }
+        ],
+        "ratio": "16:9",
+        "test": True,
+        "version": "v1alpha"
+    }
+    response = requests.post(CREATE_VIDEO_URL, headers=POST_HEADER, json=data)
+    print(response.status_code)
+    print(response.json())
+
+    # GET TO CHECK STATUS AND GET VIDEO WHEN READY
+    video_id = response.json()["data"]["video_id"]
+    params = {
+        'video_id': response.json()["data"]["video_id"]
+    }
+    response = {}
+    status = "processing"
+    error = None
+
+    while status != "completed" and error is None:
+        response = requests.get(GET_VIDEO_URL, headers=GET_HEADER, params=params)
+        response_data = response.json()
+
+        status = response_data["data"]["status"]
+        error = response_data["data"]["error"]
+
+        if status != "completed" and error is None:
+            print(f"Status: {status}")
+            time.sleep(5)  # Wait for 5 second before the next request
+
+    print(response.status_code)
+    print(response.json())
+
+    # DOWNLOAD VIDEO
+    download_url = response.json()['data']['video_url']
+    output_directory = 'download-video/'
+    output_filename = video_id + '.mp4'
+
+    response = requests.get(download_url)
+
+    if response.status_code == 200:
+        os.makedirs(output_directory, exist_ok=True)  # Create the directory if it doesn't exist
+        output_path = os.path.join(output_directory, output_filename)
+        with open(output_path, 'wb') as f:
+            f.write(response.content)
+        print(f"File '{output_filename}' downloaded successfully.")
+        return output_filename
+    else:
+        print(f"Failed to download file. Status code: {response.status_code}")
+        return None