Merge pull request #1 from f-aguzzi/main

Refactoring of text to speech function.

text_to_speech.py

@@ -1,28 +1,20 @@
-"""
-Basic example of scraping pipeline using SpeechSummaryGraph
-"""
-
 from scrapegraphai.graphs import SpeechSummaryGraph
 
-def text_to_speech(key:str, prompt:str, url:str):
-    """ 
-    Given a text it reads after the prompt
+def text_to_speech(api_key: str, prompt: str, url: str):
+    """Reads text after the prompt from a given URL.
+
     Args:
-        - url (str): url to scrape
-        - prompt(str): prompt to do
-        - key(str) openaikey
+        - api_key (str): OpenAI API key
+        - prompt (str): Prompt to use
+        - url (str): URL to scrape
+    Returns:
+        - str: Path to the generated audio file
     """
-
-
-    llm_config = {
-        "api_key": key
-    }
-
-    # Save the audio to a file
+    llm_config = {"api_key": api_key}
+
+    # Define the name of the audio file
     audio_file = "audio_result.mp3"
-    speech_summary_graph = SpeechSummaryGraph(prompt,
-                                url, llm_config,
-                                    audio_file)
 
-    return speech_summary_graph.run()
-
+    # Create and run the speech summary graph
+    speech_summary_graph = SpeechSummaryGraph(prompt, url, llm_config, audio_file)
+    return speech_summary_graph.run()