Upload server.py

server.py

@@ -39,31 +39,6 @@ def get_embedding_function():
     return embeddings
 
 
-@app.get("/")
-def greet_json():
-    return {"Hello": "World!"}
-
-
-class QueryRequest(BaseModel):
-    query_text: str
-
-
-@app.get("/train")
-def train():
-
-    # Check if the database should be cleared (using the --clear flag).
-    parser = argparse.ArgumentParser()
-    parser.add_argument("--reset", action="store_true", help="Reset the database.")
-    args = parser.parse_args()
-    if args.reset:
-        print("✨ Clearing Database")
-        clear_database()
-
-    # Create (or update) the data store.
-    documents = load_documents()
-    chunks = split_documents(documents)
-    add_to_chroma(chunks)
-
 
 def load_documents():
     document_loader = PyPDFDirectoryLoader(DATA_PATH)
@@ -145,6 +120,35 @@ def clear_database():
     return {""}
 
 
+
+@app.get("/")
+def greet_json():
+    return {"Hello": "World!"}
+
+
+class QueryRequest(BaseModel):
+    query_text: str
+
+
+@app.get("/train")
+def train():
+
+    # Check if the database should be cleared (using the --clear flag).
+    parser = argparse.ArgumentParser()
+    parser.add_argument("--reset", action="store_true", help="Reset the database.")
+    args = parser.parse_args()
+    if args.reset:
+        print("✨ Clearing Database")
+        clear_database()
+
+    # Create (or update) the data store.
+    documents = load_documents()
+    chunks = split_documents(documents)
+    add_to_chroma(chunks)
+
+
+
+
 @app.get("/query")
 def query(request: QueryRequest):
     query_text = request.query_text
@@ -168,17 +172,3 @@ def query(request: QueryRequest):
     print(formatted_response)
     return response_text
 
-
-
-
-
-
-
-
-
-
-
-
-
-
-