Spaces:
Runtime error
Runtime error
Walid Aissa
commited on
Commit
·
22eefa0
1
Parent(s):
e8d113c
Improved the wikipedia_search method
Browse files
app.py
CHANGED
|
@@ -42,7 +42,14 @@ def wikipedia_search(input: str) -> str:
|
|
| 42 |
|
| 43 |
try :
|
| 44 |
#TODO: add better extraction and search
|
| 45 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 46 |
return page.summary
|
| 47 |
except:
|
| 48 |
return "I cannot answer this question"
|
|
|
|
| 42 |
|
| 43 |
try :
|
| 44 |
#TODO: add better extraction and search
|
| 45 |
+
keyphrase_index = 0
|
| 46 |
+
page = wiki.page(keyphrases[keyphrase_index])
|
| 47 |
+
|
| 48 |
+
while not ('.' in page.summary) or not page.exists():
|
| 49 |
+
keyphrase_index += 1
|
| 50 |
+
if keyphrase_index == len(keyphrases):
|
| 51 |
+
raise Exception
|
| 52 |
+
page = wiki.page(keyphrases[keyphrase_index])
|
| 53 |
return page.summary
|
| 54 |
except:
|
| 55 |
return "I cannot answer this question"
|