switched to fastapi

main.py

@@ -25,16 +25,22 @@ def hello():
     return {"massage": "we are running!"}
 
 
-@app.get("/generate/")
-def generate(prompt: str, lang: str):
-    # request_data = prompt.dict()
+@app.get("/test")
+def test(input: str):
+    output = input
+    return {"input": output}
 
-    # text = request_data["prompts"]
 
-    if lang:
+@app.post("/generate/")
+def generate(prompt: Prompt):
+    request_data = prompt.dict()
+
+    text = request_data["prompts"]
+
+    if request_data.get("lang"):
         generated_recepie = generation_function(
-            prompt, lang=lang)
+            text, lang=request_data["lang"])
     else:
-        generated_recepie = generation_function(prompt)
+        generated_recepie = generation_function(text)
 
     return generated_recepie