Update app.py

app.py

@@ -46,6 +46,7 @@ def bot(
     google_key: str,
     model_name: str,
     image_prompt: Optional[Image.Image],
+    video_prompt,
     temperature: float,
     max_output_tokens: int,
     stop_sequences: str,
@@ -84,6 +85,23 @@ def bot(
             generation_config=generation_config)
         response.resolve()
 
+    if video_prompt is None:
+        model = genai.GenerativeModel(model_name)
+        response = model.generate_content(
+            text_prompt,
+            stream=True,
+            generation_config=generation_config)
+        response.resolve()
+    else:
+        video_prompt = preprocess_image(video_prompt)
+        model = genai.GenerativeModel('gemini-pro-vision')
+        response = model.generate_content(
+            contents=[text_prompt, video_prompt_prompt],
+            stream=True,
+            generation_config=generation_config)
+        response.resolve()
+
+    
     # streaming effect
     chatbot[-1][1] = ""
     for chunk in response:
@@ -103,6 +121,7 @@ google_key_component = gr.Textbox(
 )
 
 image_prompt_component = gr.Image(type="pil", label="Image", scale=1)
+video_prompt_component = gr.Video()
 
 model_selection = gr.Dropdown(["gemini-1.0-pro", "gemini-pro-vision","gemini-1.5-flash-latest", "gemini-1.5-pro-latest","gemini-1.0-pro-001"],label="Select Gemini Model",value="gemini-1.0-pro")
 
@@ -182,6 +201,7 @@ bot_inputs = [
     google_key_component,
     model_selection, 
     image_prompt_component,
+    video_prompt_component,
     temperature_component,
     max_output_tokens_component,
     stop_sequences_component,
@@ -198,6 +218,7 @@ with gr.Blocks() as demo:
         google_key_component.render()
         with gr.Row():
             image_prompt_component.render()
+            video_prompt_component.render()
             model_selection.render()
             chatbot_component.render()
         text_prompt_component.render()