Update app.py

app.py

@@ -2,6 +2,7 @@ import os
 import time
 from typing import List, Tuple, Optional
 from pathlib import Path
+from moviepy.editor import VideoFileClip, concatenate_videoclips
 
 import google.generativeai as genai
 import gradio as gr
@@ -42,6 +43,7 @@ def user(text_prompt: str, chatbot: List[Tuple[str, str]]):
     return "", chatbot + [[text_prompt, None]]
 
 
+
 def bot(
     google_key: str,
     model_name: str,
@@ -71,45 +73,29 @@ def bot(
         top_k=top_k,
         top_p=top_p)
 
-    if video_prompt is not None and image_prompt is not None:
-        model = genai.GenerativeModel(model_name)
-
-        image_prompt = preprocess_image(image_prompt)
-        for video_path in video_prompt:
-            video_file = genai.upload_file(path=video_path)
-            while video_file.state.name == "PROCESSING":
-                print('.', end='')
-                time.sleep(10)
-                video_file = genai.get_file(video_file.name)
-            
-            if video_file.state.name == "FAILED":
-                raise ValueError(video_file.state.name)
+    if video_prompt is not None:
+        video_clips = [VideoFileClip(str(video_path)) for video_path in video_prompt]
+        merged_video = concatenate_videoclips(video_clips)
+        merged_video_path = "merged_video.mp4"
+        merged_video.write_videofile(merged_video_path)
 
-            response = model.generate_content(
-                contents=[video_file, image_prompt, text_prompt],
-                stream=True,
-                generation_config=generation_config,
-                request_options={"timeout": 600})
-            response.resolve()
-    elif video_prompt is not None:
         model = genai.GenerativeModel(model_name)
 
-        for video_path in video_prompt:
-            video_file = genai.upload_file(path=video_path)
-            while video_file.state.name == "PROCESSING":
-                print('.', end='')
-                time.sleep(10)
-                video_file = genai.get_file(video_file.name)
-            
-            if video_file.state.name == "FAILED":
-                raise ValueError(video_file.state.name)
+        video_file = genai.upload_file(path=merged_video_path)
+        while video_file.state.name == "PROCESSING":
+            print('.', end='')
+            time.sleep(10)
+            video_file = genai.get_file(video_file.name)
+        
+        if video_file.state.name == "FAILED":
+            raise ValueError(video_file.state.name)
 
-            response = model.generate_content(
-                contents=[video_file, text_prompt],
-                stream=True,
-                generation_config=generation_config,
-                request_options={"timeout": 600})
-            response.resolve()        
+        response = model.generate_content(
+            contents=[video_file, text_prompt],
+            stream=True,
+            generation_config=generation_config,
+            request_options={"timeout": 600})
+        response.resolve()        
     elif image_prompt is not None:
         image_prompt = preprocess_image(image_prompt)
         model = genai.GenerativeModel(model_name)