Update app.py

app.py

@@ -56,19 +56,19 @@ def input_image_setup(file_loc):
         ]
     return image_parts
 
-def generate_gemini_response(input_prompt, image_loc):
+def generate_gemini_response(input_prompt,, image_loc):
     image_prompt = input_image_setup(image_loc)
-    prompt_parts = [input_prompt, 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 """
+input_prompt = """ give the plate, the model and the color of the car """
 
-def upload_file(files):
+def upload_file(files, social_media_platform, num_captions=1):
     if not files:
         return None, "Image not uploaded"
     file_paths = [file.name for file in files]
-    response = generate_gemini_response(input_prompt, file_paths[0])
+    response = generate_gemini_response(input_prompt + f"\nPlatform: {social_media_platform}\n"+ f"Number of Captions/Posts: {num_captions}\n", file_paths[0])
     return file_paths[0], response
 
 with gr.Blocks() as demo: