Update app.py

app.py

@@ -46,7 +46,39 @@ if not OPENAI_API_KEY:
 
 openai.api_key = OPENAI_API_KEY
 
-# ... (keep all the helper functions as they are) ...
+# Add this function definition after the other helper functions and before the app layout
+
+def process_media(contents, filename, url):
+    logger.info("Starting media processing")
+    try:
+        if contents:
+            content_type, content_string = contents.split(',')
+            decoded = base64.b64decode(content_string)
+            suffix = os.path.splitext(filename)[1]
+            with tempfile.NamedTemporaryFile(delete=False, suffix=suffix) as temp_file:
+                temp_file.write(decoded)
+                temp_file_path = temp_file.name
+            logger.info(f"File uploaded: {temp_file_path}")
+        elif url:
+            temp_file_path = download_media(url)
+        else:
+            logger.error("No input provided")
+            raise ValueError("No input provided")
+
+        if temp_file_path.lower().endswith(('.mp4', '.avi', '.mov', '.flv', '.wmv')):
+            logger.info("Video file detected, extracting audio")
+            audio_file_path = extract_audio(temp_file_path)
+            transcript = transcribe_audio(audio_file_path)
+            os.unlink(audio_file_path)
+        else:
+            logger.info("Audio file detected, transcribing directly")
+            transcript = transcribe_audio(temp_file_path)
+
+        os.unlink(temp_file_path)
+        return transcript
+    except Exception as e:
+        logger.error(f"Error in process_media: {str(e)}")
+        raise
 
 app.layout = dbc.Container([
     dbc.Row([