Update app.py

app.py

@@ -171,11 +171,7 @@ def generate_prompt(query: str, context_snippets: list) -> str:
 #     return answer
 
 def get_gemini_flash_response(prompt: str) -> str:
-    import google.generativeai as genai
-    from google.generativeai import palm
-
-
-    
+    import google.generativeai as palm  # Import as palm
     palm.configure(api_key=GEMINI_API_KEY)
     
     response = palm.generate_text(
@@ -190,6 +186,7 @@ def get_gemini_flash_response(prompt: str) -> str:
 
 
 
+
 ############################################
 # Gradio Interface Functions
 ############################################