Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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("
|
| 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"
|
| 100 |
-
return "
|
| 101 |
if not self.gemini_api_key:
|
| 102 |
-
return "
|
| 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"
|
| 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://
|
| 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():
|