Blocking other spaces

TextGen/router.py

@@ -168,25 +168,27 @@ def inference_model(system_messsage, prompt):
 
 @app.get("/", tags=["Home"])
 def api_home(request: Request):
-    client_ip = request.client.host
-    print(client_ip)
+    if request.headers.get("origin") != "https://jofthomas-everchanging-quest.static.hf.space":
+        return
     return {'detail': 'Everchanging Quest backend, nothing to see here'}
 
 @app.post("/api/generate", summary="Generate text from prompt", tags=["Generate"], response_model=Generate)
-def inference(message: Message):
+def inference(message: Message, request: Request):
+    if request.headers.get("origin") != "https://jofthomas-everchanging-quest.static.hf.space":
+        return
     return generate_text(messages=message.messages, npc=message.npc)
 
 @app.post("/invoke_model")
-def story(prompt: Invoke):
+def story(prompt: Invoke,request: Request):
+    if request.headers.get("origin") != "https://jofthomas-everchanging-quest.static.hf.space":
+        return
     return inference_model(system_messsage=prompt.system_prompt,prompt=prompt.message)
     
 @app.post("/generate_level")
 def placement(input: Rooms, request: Request):
-    origin = request.headers.get("origin")
-    referer = request.headers.get("referer")
-    print("origin : ", origin)
-    print("referer : ", referer)
-    print(input)
+    print(request.headers)
+    if request.headers.get("origin") != "https://jofthomas-everchanging-quest.static.hf.space":
+        return
     markdown_map=generate_map_markdown(input.rooms,input.room_of_interest,input.index_exit)
     story=generate_story(input.possible_entities)
     print(story)
@@ -195,7 +197,9 @@ def placement(input: Rooms, request: Request):
     return placements
     
 @app.post("/check_right_to_pass")
-def check(input: Logs):
+def check(input: Logs,request: Request):
+    if request.headers.get("origin") != "https://jofthomas-everchanging-quest.static.hf.space":
+        return
     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"
@@ -259,7 +263,9 @@ async def generate_wav(message: VoiceMessage):
 
 @app.get("/generate_voice_eleven", response_class=StreamingResponse)
 @app.post("/generate_voice_eleven", response_class=StreamingResponse)
-def generate_voice_eleven(message: VoiceMessage = None):
+def generate_voice_eleven(message: VoiceMessage = None,request: Request):
+    if request.headers.get("origin") != "https://jofthomas-everchanging-quest.static.hf.space":
+        return
     global Last_message  # Declare Last_message as global
     if message is None:
         message = Last_message
@@ -308,7 +314,9 @@ def generate_voice_eleven(message: VoiceMessage = None):
 #    return StreamingResponse(audio_stream(),media_type="audio/mpeg")  
 @app.post("/generate_song")
 @app.get("/generate_song")
-async def generate_song(request:SongRequest):
+async def generate_song(request:SongRequest,request: Request):
+    if request.headers.get("origin") != "https://jofthomas-everchanging-quest.static.hf.space":
+        return
     backstory=main_npc_system_prompts["Bard"]
     print(backstory)
     text=f"""the story is about a little girl in red hood adventuring in the dungeon behind the portal.{backstory} /n The user requested a song about : {request.prompt}"""