Update app.py

app.py

@@ -10,17 +10,17 @@ from pydantic import BaseModel
 from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
 from sentence_transformers import SentenceTransformer, models
 import gradio as gr
+import os
 
 ############################################
 # Configuration
 ############################################
 
-import os
 
 HF_TOKEN = os.environ.get("HF_TOKEN")
 GITHUB_TOKEN = os.environ.get("GITHUB_TOKEN")
 GEMINI_API_KEY = os.environ.get("GEMINI_API_KEY")
-
+os.system("pip install -q -U google-genai")
 
 
 ############################################
@@ -172,6 +172,7 @@ def generate_prompt(query: str, context_snippets: list) -> str:
 
 def get_gemini_flash_response(prompt: str) -> str:
     from google import genai 
+    from google.genai import types
     client = genai.Client(api_key=GEMINI_API_KEY)