Add FastAPI

app.py

@@ -7,6 +7,9 @@ from llama_index import (
 )
 import chromadb
 from typing import TypedDict
+from fastapi import FastAPI, File, UploadFile, Request
+from fastapi.responses import JSONResponse
+
 
 from llama_index.llms import MistralAI
 from llama_index.embeddings import MistralAIEmbedding
@@ -48,6 +51,8 @@ index = VectorStoreIndex(
 )
 query_engine = index.as_query_engine(similarity_top_k=5)
 
+app = FastAPI()
+
 # Structure of the data sent by the form
 InputForm = TypedDict('InputForm', {
   'department': str,
@@ -58,8 +63,9 @@ InputForm = TypedDict('InputForm', {
 })
 
 # This function is the API endpoint the web app will use 
+@app.post("/find-my-rotation")   
 def find_my_rotation(input: InputForm):
-    return "Hi, you. Your farm is " + input["farmSize"] + "ha"
+    return JSONResponse({"response":  "Hi, you. Your farm is " + input["farmSize"] + "ha"})
 
 
 def get_documents_in_db():