Spaces:
Build error
Build error
Upload server.py
Browse files
server.py
CHANGED
|
@@ -134,12 +134,12 @@ class QueryRequest(BaseModel):
|
|
| 134 |
def train():
|
| 135 |
|
| 136 |
# Check if the database should be cleared (using the --clear flag).
|
| 137 |
-
parser = argparse.ArgumentParser()
|
| 138 |
-
parser.add_argument("--reset", action="store_true", help="Reset the database.")
|
| 139 |
-
args = parser.parse_args()
|
| 140 |
-
if args.reset:
|
| 141 |
-
print("✨ Clearing Database")
|
| 142 |
-
clear_database()
|
| 143 |
|
| 144 |
# Create (or update) the data store.
|
| 145 |
documents = load_documents()
|
|
@@ -149,6 +149,21 @@ def train():
|
|
| 149 |
|
| 150 |
|
| 151 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 152 |
@app.get("/query")
|
| 153 |
def query(request: QueryRequest):
|
| 154 |
query_text = request.query_text
|
|
|
|
| 134 |
def train():
|
| 135 |
|
| 136 |
# Check if the database should be cleared (using the --clear flag).
|
| 137 |
+
# parser = argparse.ArgumentParser()
|
| 138 |
+
# parser.add_argument("--reset", action="store_true", help="Reset the database.")
|
| 139 |
+
# args = parser.parse_args()
|
| 140 |
+
# if args.reset:
|
| 141 |
+
# print("✨ Clearing Database")
|
| 142 |
+
# clear_database()
|
| 143 |
|
| 144 |
# Create (or update) the data store.
|
| 145 |
documents = load_documents()
|
|
|
|
| 149 |
|
| 150 |
|
| 151 |
|
| 152 |
+
@app.get("/cleardb")
|
| 153 |
+
def cleardb():
|
| 154 |
+
|
| 155 |
+
# Check if the database should be cleared (using the --clear flag).
|
| 156 |
+
# parser = argparse.ArgumentParser()
|
| 157 |
+
# parser.add_argument("--reset", action="store_true", help="Reset the database.")
|
| 158 |
+
# args = parser.parse_args()
|
| 159 |
+
# if args.reset:
|
| 160 |
+
print("✨ Clearing Database")
|
| 161 |
+
clear_database()
|
| 162 |
+
|
| 163 |
+
|
| 164 |
+
|
| 165 |
+
|
| 166 |
+
|
| 167 |
@app.get("/query")
|
| 168 |
def query(request: QueryRequest):
|
| 169 |
query_text = request.query_text
|