Update app.py

app.py

@@ -171,8 +171,9 @@ def generate_prompt(query: str, context_snippets: list) -> str:
 #     return answer
 
 def get_gemini_flash_response(prompt: str) -> str:
-    from google import genai
-    from google.genai import types
+    import google.generativeai as genai
+    from google.generativeai import types
+
 
     # Create a Gemini client using the API key from the environment.
     client = genai.Client(api_key=GEMINI_API_KEY)