Update TextGen/router.py

TextGen/router.py

@@ -11,7 +11,7 @@ from langchain.prompts import PromptTemplate
 from TextGen.suno import custom_generate_audio, get_audio_information,generate_lyrics
 from TextGen.gemini import generate_story,place_objects,generate_map_markdown
 #from TextGen.diffusion import generate_image
-#from coqui import predict
+from coqui import predict
 from langchain_google_genai import (
     ChatGoogleGenerativeAI,
     HarmBlockThreshold,
@@ -168,27 +168,27 @@ 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 204
+    #if request.headers.get("origin") != "https://jofthomas-everchanging-quest.static.hf.space":
+    #    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 204
+    #if request.headers.get("origin") != "https://jofthomas-everchanging-quest.static.hf.space":
+    #    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 204
+    #if request.headers.get("origin") != "https://jofthomas-everchanging-quest.static.hf.space":
+    #    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 204
+    #if request.headers.get("origin") != "https://jofthomas-everchanging-quest.static.hf.space":
+    #    return 204
     markdown_map=generate_map_markdown(input.rooms,input.room_of_interest,input.index_exit)
     story=generate_story(input.possible_entities)
     print(story)
@@ -198,8 +198,8 @@ 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 204
+    #if request.headers.get("origin") != "https://jofthomas-everchanging-quest.static.hf.space":
+    #    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"
@@ -261,62 +261,62 @@ async def generate_wav(message: VoiceMessage):
     return 200
 
 
-@app.get("/generate_voice_eleven", response_class=StreamingResponse)
-@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 204
-    global Last_message  # Declare Last_message as global
-    if message is None:
-        message = Last_message
-    else:
-        Last_message = message
-
-    def audio_stream():
-        this_voice_id=determine_elevenLav_voice_from_npc(message.npc, message.genre)
-        
-        # Generate the audio stream from ElevenLabs
-        for chunk in Eleven_client.generate(text=message.input,
-                                            voice=Voice(
-                                                voice_id=this_voice_id,
-                                                settings=VoiceSettings(stability=0.71, similarity_boost=0.5, style=0.0, use_speaker_boost=True)
-                                            ),
-                                            stream=True):
-            yield chunk
-
-    return StreamingResponse(audio_stream(), media_type="audio/mpeg")
-#@app.get("/generate_voice_coqui", response_class=StreamingResponse)
-#@app.post("/generate_voice_coqui", response_class=StreamingResponse)
-#def generate_voice_coqui(message: VoiceMessage = None):
-#    global Last_message 
+#@app.get("/generate_voice_eleven", response_class=StreamingResponse)
+#@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 204
+#    global Last_message  # Declare Last_message as global
 #    if message is None:
 #        message = Last_message
 #    else:
 #        Last_message = message
 #
 #    def audio_stream():
-#        voice = determine_vocie_from_npc(message.npc, message.genre)
-#        result = predict(
-#                prompt=message.input,
-#                language=message.language,
-#                audio_file_pth=voice,
-#                mic_file_path=None,
-#                use_mic=False,
-#                voice_cleanup=False,
-#                no_lang_auto_detect=False,
-#                agree=True,
-#            )
+#        this_voice_id=determine_elevenLav_voice_from_npc(message.npc, message.genre)
+#        
 #        # Generate the audio stream from ElevenLabs
-#        for chunk in result:
-#            print("received : ",chunk)
-#            yield chunk#
-#
-#    return StreamingResponse(audio_stream(),media_type="audio/mpeg")  
+###        for chunk in Eleven_client.generate(text=message.input,
+#                                            voice=Voice(
+#                                                voice_id=this_voice_id,
+#                                                settings=VoiceSettings(stability=0.71, similarity_boost=0.5, style=0.0, use_speaker_boost=True)
+#                                            ),
+#                                            stream=True):
+#            yield chunk
+
+ #   return StreamingResponse(audio_stream(), media_type="audio/mpeg")
+@app.get("/generate_voice_coqui", response_class=StreamingResponse)
+@app.post("/generate_voice_coqui", response_class=StreamingResponse)
+def generate_voice_coqui(message: VoiceMessage = None):
+    global Last_message 
+    if message is None:
+        message = Last_message
+    else:
+        Last_message = message
+
+    def audio_stream():
+        voice = determine_vocie_from_npc(message.npc, message.genre)
+        result = predict(
+                prompt=message.input,
+                language=message.language,
+                audio_file_pth=voice,
+                mic_file_path=None,
+                use_mic=False,
+                voice_cleanup=False,
+                no_lang_auto_detect=False,
+                agree=True,
+            )
+        # Generate the audio stream from ElevenLabs
+        for chunk in result:
+            print("received : ",chunk)
+            yield chunk#
+
+    return StreamingResponse(audio_stream(),media_type="audio/mpeg")  
 @app.post("/generate_song")
 @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 204
+    #if httprequest.headers.get("origin") != "https://jofthomas-everchanging-quest.static.hf.space":
+    #    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}"""