Update TextGen/router.py

TextGen/router.py

@@ -62,6 +62,10 @@ class VoiceMessage(BaseModel):
     input: str | None = None
     language: str | None = "en"
     genre:str | None = "Male"
+
+class Logs(BaseModel):
+    objective:str
+    logs: List
     
 song_base_api=os.environ["VERCEL_API"]
 
@@ -188,6 +192,14 @@ def placement(input: Rooms):
     placements=place_objects(input.possible_entities,story,markdown_map)
     print(placements)
     return placements
+    
+@app.post("/check_right_to_pass")
+def check(input: Logs):
+    print(input.logs)
+    system_prompt="You are a game master in a roguelike. You previously decided of an objective for the player. You have to answer with YES or NO on wether the objective as been sucessfully completed. Be kind and flexible as the content of the game is AI generated and might not be feasible."
+    user_message=f"The objective was : {input.objective} and the player did the following actions : {input.logs}. Do you grant acess ? ONLY answer YES or NO"
+    answer=inference_model(system_prompt,user_message)
+    return answer
 
 #Dummy function for now
 def determine_vocie_from_npc(npc,genre):