Add find_my_rotation function

app.py

@@ -6,6 +6,7 @@ from llama_index import (
     download_loader,
 )
 import chromadb
+from typing import TypedDict
 
 from llama_index.llms import MistralAI
 from llama_index.embeddings import MistralAIEmbedding
@@ -47,6 +48,19 @@ index = VectorStoreIndex(
 )
 query_engine = index.as_query_engine(similarity_top_k=5)
 
+# Structure of the data sent by the form
+InputForm = TypedDict('InputForm', {
+  'department': str,
+  'farmSize': float,
+  'benefitsFromCommonAgriculturalPolicy': bool,
+  'cultures': list[str],
+  'yields': dict[str, float]
+})
+
+# This function is the API endpoint the web app will use 
+def find_my_rotation(input: InputForm):
+    return "Hi, you. Your farm is " + input["farmSize"] + "ha"
+
 
 def get_documents_in_db():
     print("Fetching documents in DB")