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
|