Update app.py

app.py

@@ -171,14 +171,18 @@ def generate_prompt(query: str, context_snippets: list) -> str:
 #     return answer
 
 def get_gemini_flash_response(prompt: str) -> str:
-    import google.generativeai as palm  # Import as palm
-    palm.configure(api_key=GEMINI_API_KEY)
+    from google import genai 
+    client = genai.Client(api_key=GEMINI_API_KEY)
+
+
     
-    response = palm.generate_text(
+    response = client.models.generate_content(
         model="gemini-2.0-flash",
-        prompt=prompt,
-        temperature=0.1,        # Adjust as needed
-        max_output_tokens=500   # Adjust as needed
+        contents=prompt,
+        config=types.GenerateContentConfig(
+        max_output_tokens=500,
+        temperature=0.1
+        )
     )
     
     return response.text