Update app.py

app.py

@@ -10,11 +10,11 @@ genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
 
 # Function to get response from Gemini model
 def get_gemini_response(input_prompt, image_data, genre, length, language, mood):
+    # Prepare the prompts
     if image_data:
         full_prompt = f"""
         You are a famous creative writer. Look at the image provided and create a {length} {genre} in {language}. The {genre} should be {mood}. The {genre} should be based on the image and contain realistic and emotional elements.
         """
-        response = genai.generate_content([full_prompt, image_data[0]])
     else:
         full_prompt = f"""
         You are a creative writer. Create a {length} {genre} in {language}. The {genre} should be {mood}. The {genre} should be based on the following prompt:
@@ -23,11 +23,13 @@ def get_gemini_response(input_prompt, image_data, genre, length, language, mood)
         
         Make sure it contains realistic and emotional elements.
         """
-        response = genai.generate_content([full_prompt])
-    
+
+    # Call the API (the method might differ; check the documentation)
+    response = genai.generate([full_prompt])  # Cambiar aquí si es necesario según la documentación
+
     # Check if response is valid and return text
-    if response and response.parts:
-        return response.parts[0].text
+    if response and response[0]:  # Ajusta el índice según la respuesta de la API
+        return response[0].text
     else:
         raise ValueError("Sorry, But please try a different combination of inputs.")