Update app.py

app.py

@@ -56,19 +56,19 @@ def input_image_setup(file_loc):
         ]
     return image_parts
 
-def generate_gemini_response(input_prompt, text_input, image_loc):
+def generate_gemini_response(input_prompt, image_loc):
     image_prompt = input_image_setup(image_loc)
-    prompt_parts = [input_prompt + text_input, image_prompt[0]]
+    prompt_parts = [input_prompt, image_prompt[0]]
     response = model.generate_content(prompt_parts)
     return response.text
 
 input_prompt = """ Give the plate and the color of the car """
 
-def upload_file(files, text_input, social_media_platform, num_captions=1):
+def upload_file(files):
     if not files:
         return None, "Image not uploaded"
     file_paths = [file.name for file in files]
-    response = generate_gemini_response(input_prompt + f"Number of Captions/Posts: {num_captions}\n", text_input, file_paths[0])
+    response = generate_gemini_response(input_prompt, file_paths[0])
     return file_paths[0], response
 
 with gr.Blocks() as demo:
@@ -82,7 +82,7 @@ with gr.Blocks() as demo:
     def process_generate(files):
         if not files:
             return None, "Image not uploaded"
-        return upload_file(files, text_input, social_media_input, num_captions_input)
+        return upload_file(files)
     
     upload_button.upload(fn=lambda files: files[0].name if files else None, inputs=[upload_button], outputs=image_output)
     generate_button.click(fn=process_generate, inputs=[upload_button], outputs=[image_output, file_output])