Update app.py

app.py

@@ -1,6 +1,7 @@
 import os
 import time
 from typing import List, Tuple, Optional
+from pathlib import Path
 
 import google.generativeai as genai
 import gradio as gr
@@ -46,6 +47,7 @@ def bot(
     model_name: str,
     image_prompt: Optional[Image.Image],
     video_prompt,
+    file_prompt,
     temperature: float,
     max_output_tokens: int,
     stop_sequences: str,
@@ -124,7 +126,8 @@ google_key_component = gr.Textbox(
 )
 
 image_prompt_component = gr.Image(type="pil", label="Image", scale=1)
-video_prompt_component = gr.Video()
+video_prompt_component = gr.Video(label="Video")
+file_prompt_component = gr.UploadButton("Upload a file", file_count="directory", label="File")
 
 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")
 
@@ -205,6 +208,7 @@ bot_inputs = [
     model_selection, 
     image_prompt_component,
     video_prompt_component,
+    file_prompt_component,
     temperature_component,
     max_output_tokens_component,
     stop_sequences_component,
@@ -223,6 +227,7 @@ with gr.Blocks() as demo:
         with gr.Row():
             image_prompt_component.render()
             video_prompt_component.render()
+            file_prompt_component.render()
             model_selection.render()
             chatbot_component.render()
         text_prompt_component.render()