Update app.py

app.py

@@ -172,29 +172,24 @@ def generate_prompt(query: str, context_snippets: list) -> str:
 
 def get_gemini_flash_response(prompt: str) -> str:
     import google.generativeai as genai
-    from google.generativeai import types
+    from google.generativeai import palm
 
 
-    # Create a Gemini client using the API key from the environment.
-    client = genai.Client(api_key=GEMINI_API_KEY)
     
-    # Configure generation settings.
-    config = types.GenerateContentConfig(
-        max_output_tokens=500,  # Adjust as needed.
-        temperature=0.1         # Lower temperature for more deterministic responses.
-    )
+    palm.configure(api_key=GEMINI_API_KEY)
     
-    # Send the prompt to the Gemini-2.0-flash model.
-    response = client.models.generate_content(
+    response = palm.generate_text(
         model="gemini-2.0-flash",
-        contents=[prompt],
-        config=config
+        prompt=prompt,
+        temperature=0.1,        # Adjust as needed
+        max_output_tokens=500   # Adjust as needed
     )
     
     return response.text
 
 
 
+
 ############################################
 # Gradio Interface Functions
 ############################################