gmustafa413 commited on
Commit
e0c46c5
·
verified ·
1 Parent(s): 911a038

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -24,7 +24,7 @@ class GeminiRAGSystem:
24
  self.chunks = []
25
  self.dataset_loaded = False
26
  self.loading_error = None
27
- self.gemini_api_key = os.getenv("GEMINI_API_KEY")
28
 
29
  # Initialize embedding model
30
  try:
@@ -96,10 +96,10 @@ class GeminiRAGSystem:
96
  """Generate response with robust error handling"""
97
  if not self.dataset_loaded:
98
  if self.loading_error:
99
- return f"⚠️ Dataset loading failed: {self.loading_error}"
100
- return "⚠️ Dataset is still loading, please wait..."
101
  if not self.gemini_api_key:
102
- return "🔑 Please set your Gemini API key in environment variables"
103
 
104
  context = self.get_relevant_context(query)
105
  if not context:
@@ -116,7 +116,7 @@ class GeminiRAGSystem:
116
  response = model.generate_content(prompt)
117
  return response.text
118
  except Exception as e:
119
- return f"⚠️ API Error: {str(e)}"
120
 
121
  # Initialize system
122
  try:
@@ -130,7 +130,7 @@ with gr.Blocks(title="UE Chatbot") as app:
130
 
131
  with gr.Row():
132
  chatbot = gr.Chatbot(height=500, label="Chat History",
133
- avatar_images=(None, (None, "https://huggingface.co/spaces/groq/Groq-LLM/resolve/main/groq_logo.png")),
134
  bubble_full_width=False)
135
 
136
  with gr.Row():
 
24
  self.chunks = []
25
  self.dataset_loaded = False
26
  self.loading_error = None
27
+ self.gemini_api_key = os.getenv("AIzaSyASrFvE3gFPigihza0JTuALzZmBx0Kc3d0")
28
 
29
  # Initialize embedding model
30
  try:
 
96
  """Generate response with robust error handling"""
97
  if not self.dataset_loaded:
98
  if self.loading_error:
99
+ return f"Dataset loading failed: {self.loading_error}"
100
+ return "Dataset is still loading, please wait..."
101
  if not self.gemini_api_key:
102
+ return "Please set your Gemini API key in environment variables"
103
 
104
  context = self.get_relevant_context(query)
105
  if not context:
 
116
  response = model.generate_content(prompt)
117
  return response.text
118
  except Exception as e:
119
+ return f"API Error: {str(e)}"
120
 
121
  # Initialize system
122
  try:
 
130
 
131
  with gr.Row():
132
  chatbot = gr.Chatbot(height=500, label="Chat History",
133
+ avatar_images=(None, (None, "https://www.google.com/imgres?q=ue%20lahore%20uoe%20image%20logo&imgurl=https%3A%2F%2Fblogger.googleusercontent.com%2Fimg%2Fb%2FR29vZ2xl%2FAVvXsEglo3Aj2gTXX2j9LIG89CiFB3uj8hS4pkueQtWZrBpLkHk00zbeXuMU1soAxFWz0tenyRIwL7-YxTS9g14DSQnvgGFRGqoWMP2otVtbKbJvQ28FVpUiAzkScgA57_EQiVZ_FOSuS9cGvWg%2Fw1200-h630-p-k-no-nu%2FUE.Logo.jpg&imgrefurl=http%3A%2F%2Fuejauharabad.blogspot.com%2F2012%2F10%2Fue-logo.html&docid=ZJhky4S29RFw9M&tbnid=sopb9CZYLJPitM&vet=12ahUKEwjv7r322aqMAxXM_7sIHSTUAnYQM3oECE8QAA..i&w=769&h=403&hcb=2&ved=2ahUKEwjv7r322aqMAxXM_7sIHSTUAnYQM3oECE8QAA")),
134
  bubble_full_width=False)
135
 
136
  with gr.Row():