Update app.py

app.py

@@ -10,12 +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):
-    model = genai.GenerativeModel('gemini-1.5-flash')
     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 = model.generate_content([full_prompt, image_data[0]])
+        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:
@@ -24,7 +23,7 @@ def get_gemini_response(input_prompt, image_data, genre, length, language, mood)
         
         Make sure it contains realistic and emotional elements.
         """
-        response = model.generate_content([full_prompt])
+        response = genai.generate_content([full_prompt])
     
     # Check if response is valid and return text
     if response and response.parts: