Spaces:
Sleeping
Sleeping
Update
Browse files- gemini_agent.py +2 -2
gemini_agent.py
CHANGED
|
@@ -342,7 +342,7 @@ class GeminiAgent:
|
|
| 342 |
def _load_answer_bank(self):
|
| 343 |
"""Load the answer bank from JSON file."""
|
| 344 |
try:
|
| 345 |
-
ans_bank_path = os.path.join(os.path.dirname(__file__), '
|
| 346 |
with open(ans_bank_path, 'r') as f:
|
| 347 |
self.answer_bank = json.load(f)
|
| 348 |
except Exception as e:
|
|
@@ -386,7 +386,7 @@ Response:"""
|
|
| 386 |
match_id = int(response.content.strip())
|
| 387 |
|
| 388 |
if match_id >= 0 and match_id < len(valid_questions):
|
| 389 |
-
print(f"Match found for query: {query}")
|
| 390 |
return valid_questions[match_id]['answer']
|
| 391 |
|
| 392 |
return None
|
|
|
|
| 342 |
def _load_answer_bank(self):
|
| 343 |
"""Load the answer bank from JSON file."""
|
| 344 |
try:
|
| 345 |
+
ans_bank_path = os.path.join(os.path.dirname(__file__), 'kndg_info.json')
|
| 346 |
with open(ans_bank_path, 'r') as f:
|
| 347 |
self.answer_bank = json.load(f)
|
| 348 |
except Exception as e:
|
|
|
|
| 386 |
match_id = int(response.content.strip())
|
| 387 |
|
| 388 |
if match_id >= 0 and match_id < len(valid_questions):
|
| 389 |
+
print(f"Hmm! Match found for query: {query}")
|
| 390 |
return valid_questions[match_id]['answer']
|
| 391 |
|
| 392 |
return None
|