Blocked other spaces from accessing my backend.

TextGen/router.py

@@ -169,26 +169,26 @@ def inference_model(system_messsage, prompt):
 @app.get("/", tags=["Home"])
 def api_home(request: Request):
     if request.headers.get("origin") != "https://jofthomas-everchanging-quest.static.hf.space":
-        return
+        return 204
     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, request: Request):
     if request.headers.get("origin") != "https://jofthomas-everchanging-quest.static.hf.space":
-        return
+        return 204
     return generate_text(messages=message.messages, npc=message.npc)
 
 @app.post("/invoke_model")
 def story(prompt: Invoke,request: Request):
     if request.headers.get("origin") != "https://jofthomas-everchanging-quest.static.hf.space":
-        return
+        return 204
     return inference_model(system_messsage=prompt.system_prompt,prompt=prompt.message)
     
 @app.post("/generate_level")
 def placement(input: Rooms, request: Request):
     print(request.headers)
     if request.headers.get("origin") != "https://jofthomas-everchanging-quest.static.hf.space":
-        return
+        return 204
     markdown_map=generate_map_markdown(input.rooms,input.room_of_interest,input.index_exit)
     story=generate_story(input.possible_entities)
     print(story)
@@ -199,7 +199,7 @@ def placement(input: Rooms, request: Request):
 @app.post("/check_right_to_pass")
 def check(input: Logs,request: Request):
     if request.headers.get("origin") != "https://jofthomas-everchanging-quest.static.hf.space":
-        return
+        return 204
     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"
@@ -265,7 +265,7 @@ async def generate_wav(message: VoiceMessage):
 @app.post("/generate_voice_eleven", response_class=StreamingResponse)
 def generate_voice_eleven(request: Request, message: VoiceMessage = None):
     if request.headers.get("origin") != "https://jofthomas-everchanging-quest.static.hf.space":
-        return
+        return 204
     global Last_message  # Declare Last_message as global
     if message is None:
         message = Last_message
@@ -316,7 +316,7 @@ def generate_voice_eleven(request: Request, message: VoiceMessage = None):
 @app.get("/generate_song")
 async def generate_song(request:SongRequest,httprequest: Request):
     if httprequest.headers.get("origin") != "https://jofthomas-everchanging-quest.static.hf.space":
-        return
+        return 204
     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}"""
@@ -324,16 +324,18 @@ async def generate_song(request:SongRequest,httprequest: Request):
     song_lyrics=generate_lyrics({
         "prompt": f"{text}",
         })
-    print(song_lyrics['text'])
-    data = custom_generate_audio({
-        "prompt": song_lyrics['text'],
-        "tags": "male bard",
-        "title":"Everchangin_Quest_song",
-        "wait_audio":True,
-       
-    })
-    infos=get_audio_information(f"{data[0]['id']},{data[1]['id']}")
-    return infos
+    if song_lyrics['text']:
+        data = custom_generate_audio({
+            "prompt": song_lyrics['text'],
+            "tags": "male bard",
+            "title":"Everchangin_Quest_song",
+            "wait_audio":True,
+           
+        })
+        infos=get_audio_information(f"{data[0]['id']},{data[1]['id']}")
+        return infos
+    else:
+        return 204
 
 #@app.post('/generate_image')
 #def Imagen(image:ImageGen=None):