Update app.py

app.py

@@ -6,9 +6,14 @@ import time
 import tempfile
 import base64
 import os
-
-
-
+from dotenv import load_dotenv
+from openai import OpenAI
+import assemblyai as aai
+
+# Load environment variables
+load_dotenv()
+aai.settings.api_key = os.getenv("ASSEMBLYAI_API_KEY")
+OpenAI.api_key = os.getenv("OPENAI_API_KEY")
 client = OpenAI()
 
 def extract_recent_frames(video_url, output_folder, duration=10, frames_per_second=1):
@@ -78,7 +83,26 @@ def main():
             st.error("Failed to extract frames.")
 
 #####################33
-
+def generate_description(base64_frames):
+    try:
+        prompt_messages = [
+            {
+                "role": "user",
+                "content": [
+                    "1. Generate a description for this sequence of video frames in about 90 words. Return the following: 1. List of objects in the video 2. Any restrictive content or sensitive content and if so which frame.",
+                    *map(lambda x: {"image": x, "resize": 428}, base64_frames[0::30]),
+                ],
+            },
+        ]
+        response = client.chat.completions.create(
+            model="gpt-4-vision-preview",
+            messages=prompt_messages,
+            max_tokens=3000,
+        )
+        return response.choices[0].message.content
+    except Exception as e:
+        print(f"Error in generate_description: {e}")
+        return None
 
 #########################################3333