Update main.py

main.py

@@ -1,10 +1,24 @@
 from flask import Flask
+from flask import request
 from langchain_community.llms import Ollama
 
 app = Flask(__name__)
 
 llm = Ollama(model="phi3")
 
-@app.route("/<lol>")
-def test(lol):
-    return llm.invoke(lol)
+@app.route('/completion', methods=['POST'])
+def completion():
+    """
+    {
+        user: <username>,
+        text: <text>
+    }
+    """
+    
+    message = request.get_json()
+    llm_output = llm.invoke(message['text'])
+    
+    print(llm_output)
+    return llm_output
+
+# curl -v -X POST 'http://127.0.0.1:8000/completion' --header 'Content-Type: application/json' --data '{"user": "test-user", "text": "What do you know about 3D graphics"}'