fix generation

Author: Goekdeniz-Guelmez

local_notebooklm/steps/helpers.py

@@ -184,24 +184,24 @@ def generate_speech(
     voice: str = "alloy",
     model_name: str = "tts-1",
     response_format: str = "wav",
-    output_path: str = "output.mp3"
+    output_path: str = "output"
 ):
     if isinstance(client, ElevenLabs):
-        client = ElevenLabs()
+        print(client)
         audio = client.text_to_speech.convert(
             text=text,
             voice_id=voice,
             model_id=model_name,
             output_format=response_format,
         )
-        save(audio=audio, filename=str(output_path))
+        save(audio=audio, filename=str(f"{output_path}.{response_format}"))
     else:
         with client.audio.speech.with_streaming_response.create(
             model=model_name,
             voice=voice,
             input=text,
             response_format=response_format
         ) as response:
-            response.stream_to_file(str(output_path))
+            response.stream_to_file(str(f"{output_path}.{response_format}"))
 
-    return output_path
+    return f"{output_path}.{response_format}"

local_notebooklm/steps/step4.py

@@ -56,7 +56,7 @@ def generate_speaker_audio(
 ) -> None:
     try:
         wait_for_next_step()
-        generate_speaker_audio(client=client, model_name=model_name, text=text, output_path=output_path, voice=voice, response_format=response_format)
+        generate_speech(client=client, model_name=model_name, text=text, output_path=output_path, voice=voice, response_format=response_format)
     except Exception as e:
         raise AudioGenerationError(f"Failed to generate audio: {str(e)}")
 

test.py

@@ -5,8 +5,8 @@
     config_path="config.json",
     format_type="summary",
     length="short",
-    style="academic",
-    preference="Focus on the key technical aspects, this podcast shoudl only be for Machine Learning researchers and engineers.",
+    style="normal",
+    # preference="Focus on the key technical aspects, this podcast shoudl only be for Machine Learning researchers and engineers.",
     output_dir="./examples/test_output",
     skip_to=4
 )