Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -8,11 +8,19 @@ def list_indexes():
|
|
| 8 |
try:
|
| 9 |
url = f"{API_URL}/list_indexes"
|
| 10 |
response = requests.get(url)
|
| 11 |
-
response.raise_for_status()
|
| 12 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 13 |
except requests.exceptions.RequestException as e:
|
| 14 |
print(f"Error fetching indexes: {e}")
|
| 15 |
-
return ["Error fetching indexes"]
|
| 16 |
|
| 17 |
def search_document(index_name, query, k):
|
| 18 |
url = f"{API_URL}/search/{index_name}"
|
|
|
|
| 8 |
try:
|
| 9 |
url = f"{API_URL}/list_indexes"
|
| 10 |
response = requests.get(url)
|
| 11 |
+
response.raise_for_status()
|
| 12 |
+
indexes = response.json()
|
| 13 |
+
|
| 14 |
+
# Check if the response is just ["indexes"] and replace with actual index names
|
| 15 |
+
if indexes == ["indexes"]:
|
| 16 |
+
return ["history", "medical"] # Hardcoded index names
|
| 17 |
+
elif isinstance(indexes, list) and indexes:
|
| 18 |
+
return indexes
|
| 19 |
+
else:
|
| 20 |
+
return ["No indexes available"]
|
| 21 |
except requests.exceptions.RequestException as e:
|
| 22 |
print(f"Error fetching indexes: {e}")
|
| 23 |
+
return ["Error fetching indexes"]
|
| 24 |
|
| 25 |
def search_document(index_name, query, k):
|
| 26 |
url = f"{API_URL}/search/{index_name}"
|