Update app.py

app.py

@@ -46,8 +46,7 @@ def bot(
     google_key: str,
     model_name: str,
     image_prompt: Optional[Image.Image],
-    video_prompt,
-    file_prompt,
+    video_prompt: List[Path],
     temperature: float,
     max_output_tokens: int,
     stop_sequences: str,
@@ -64,17 +63,6 @@ def bot(
 
     text_prompt = chatbot[-1][0]
 
-    if file_prompt is not None:
-        # Initialize an empty string to store content
-        all_content = ""
-        for file in file_prompt:
-            # Open the file in read binary mode ('rb')
-            with open(file.name, 'rb') as f:
-              # Read the entire file content into a byte string
-              content = f.read()
-              # Decode the byte string to a UTF-8 string (assuming text file)
-              all_content += content.decode('utf-8')
-              text_prompt = all_content + " " + chatbot[-1][0]
     genai.configure(api_key=google_key)
     generation_config = genai.types.GenerationConfig(
         temperature=temperature,
@@ -87,41 +75,41 @@ def bot(
         model = genai.GenerativeModel(model_name)
 
         image_prompt = preprocess_image(image_prompt)
-        video_file = genai.upload_file(path=video_prompt)
-        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, image_prompt, text_prompt],
-            #contents=[text_prompt, video_file],
-            stream=True,
-            generation_config=generation_config,
-            request_options={"timeout": 600})
-        response.resolve()
+        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)
+
+            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)
 
-        video_file = genai.upload_file(path=video_prompt)
-        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],
-            #contents=[text_prompt, video_file],
-            stream=True,
-            generation_config=generation_config,
-            request_options={"timeout": 600})
-        response.resolve()        
+        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)
+
+            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)
@@ -157,15 +145,8 @@ google_key_component = gr.Textbox(
     visible=GOOGLE_API_KEY is None
 )
 
-def upload_file(files):
-    file_paths = [file.name for file in files]
-    return file_paths
-
 image_prompt_component = gr.Image(type="pil", label="Image", scale=1)
-video_prompt_component = gr.Video(label="Video")
-file_prompt_component = gr.UploadButton("Click to Upload Text Files", file_count="multiple", 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")
+video_prompt_component = gr.File(label="Video", file_count="multiple")
 model_selection = gr.Dropdown(["gemini-1.5-flash-latest", "gemini-1.5-pro-latest"],label="Select Gemini Model",value="gemini-1.5-pro-latest")
 
 chatbot_component = gr.Chatbot(
@@ -217,7 +198,7 @@ top_k_component = gr.Slider(
     label="Top-K",
     info=(
         "Top-k changes how the model selects tokens for output. A top-k of 1 means the "
-        "selected token is the most probable among all tokens in the model’s "
+        "selected token is the most probable among all tokens in the model's "
         "vocabulary (also called greedy decoding), while a top-k of 3 means that the "
         "next token is selected from among the 3 most probable tokens (using "
         "temperature)."
@@ -245,7 +226,6 @@ bot_inputs = [
     model_selection, 
     image_prompt_component,
     video_prompt_component,
-    file_prompt_component,
     temperature_component,
     max_output_tokens_component,
     stop_sequences_component,
@@ -264,9 +244,6 @@ with gr.Blocks() as demo:
         with gr.Row():
             image_prompt_component.render()
             video_prompt_component.render()
-            file_prompt_component.render()
-            file_output = gr.File()
-            file_prompt_component.upload(upload_file, file_prompt_component, file_output)
             model_selection.render()
             chatbot_component.render()
         text_prompt_component.render()
@@ -308,7 +285,6 @@ with gr.Blocks() as demo:
                     "gemini-1.5-pro-latest", 
                     None,
                     None,
-                    None,
                     .4,
                     1024,
                     "",
@@ -322,7 +298,6 @@ with gr.Blocks() as demo:
                     "gemini-1.5-pro-latest", 
                     None,
                     None,
-                    None,
                     .4,
                     1024,
                     "",
@@ -336,7 +311,6 @@ with gr.Blocks() as demo:
                     "gemini-1.5-pro-latest", 
                     "./example1.webp",
                     None,
-                    None,
                     .4,
                     1024,
                     "",
@@ -350,7 +324,6 @@ with gr.Blocks() as demo:
                     "gemini-1.5-pro-latest", 
                     "./example2.jpg",
                     None,
-                    None,
                     .4,
                     1024,
                     "",
@@ -363,8 +336,7 @@ with gr.Blocks() as demo:
                     "",
                     "gemini-1.5-pro-latest", 
                     None,
-                    "./example1.mp4",
-                    None,
+                    ["./example1.mp4"],
                     .4,
                     1024,
                     "",
@@ -377,8 +349,7 @@ with gr.Blocks() as demo:
                     "",
                     "gemini-1.5-pro-latest", 
                     None,
-                    "./example2.mp4",
-                    None,
+                    ["./example2.mp4"],
                     .4,
                     1024,
                     "",
@@ -391,8 +362,7 @@ with gr.Blocks() as demo:
                     "",
                     "gemini-1.5-pro-latest", 
                     None,
-                    "./example3.mp4",
-                    None,
+                    ["./example3.mp4"],
                     .4,
                     1024,
                     "",
@@ -405,5 +375,4 @@ with gr.Blocks() as demo:
             #cache_examples="lazy",
         )
 
-demo.queue(max_size=99).launch(debug=False, show_error=True)
-
+demo.queue(max_size=99).launch(debug=False, show_error=True)