Update openai_tts_tool.py

openai_tts_tool.py

@@ -1,41 +1,45 @@
 from openai import OpenAI
 import os
 
-def generate_audio_and_text(api_key, input_text, model_name, voice_type, voice_speed, language, output_option, summary_length, additional_prompt):
+def generate_audio_and_text(api_key, input_text, model_name, voice_type, voice_speed, output_option):
+    """
+    Generate audio and text files from input text using OpenAI's TTS API.
+    
+    Args:
+        api_key (str): OpenAI API key
+        input_text (str): Text to convert to speech
+        model_name (str): OpenAI model name
+        voice_type (str): Voice type for TTS
+        voice_speed (float): Speed of speech
+        output_option (str): Output type ('audio', 'script_text', or 'both')
+    
+    Returns:
+        tuple: (audio_file_path, script_file_path, status_message)
+    """
     if not input_text:
-        return None, "No input text provided"
+        return None, None, "No input text provided"
+    
+    if not api_key:
+        return None, None, "No API key provided"
     
     try:
         client = OpenAI(api_key=api_key)
         
-        # Generate summary if requested
-        summary_text = None
-        if output_option in ["summary_text", "both"]:
-            summary_prompt = f"Summarize the following text in approximately {summary_length} words. {additional_prompt or ''}\n\nText: {input_text}"
-            
-            summary_response = client.chat.completions.create(
-                model=model_name,
-                messages=[{"role": "user", "content": summary_prompt}]
-            )
-            summary_text = summary_response.choices[0].message.content
+        # Create temp directory if it doesn't exist
+        temp_dir = os.path.join(os.getcwd(), 'temp')
+        if not os.path.exists(temp_dir):
+            os.makedirs(temp_dir)
         
-        # Generate audio if requested
+        # Generate audio file
         audio_file = None
         if output_option in ["audio", "both"]:
-            # Ensure language is passed correctly for TTS generation
             speech_response = client.audio.speech.create(
-                model="tts-1",  # Adjust model if necessary for other languages
+                model="tts-1",
                 voice=voice_type,
                 input=input_text,
-                speed=float(voice_speed),
-                language=language  # Pass the selected language here
+                speed=float(voice_speed)
             )
             
-            # Create temp directory if it doesn't exist
-            temp_dir = os.path.join(os.getcwd(), 'temp')
-            if not os.path.exists(temp_dir):
-                os.makedirs(temp_dir)
-            
             # Save the audio to a temporary file
             audio_path = os.path.join(temp_dir, f"output_{hash(input_text)}.mp3")
             with open(audio_path, "wb") as f:
@@ -44,15 +48,16 @@ def generate_audio_and_text(api_key, input_text, model_name, voice_type, voice_s
             
             audio_file = audio_path
         
-        # Return based on output option
-        if output_option == "summary_text":
-            return None, summary_text
-        elif output_option == "audio":
-            return audio_file, None
-        elif output_option == "both":
-            return audio_file, summary_text
+        # Save the input text as a script file
+        script_file = None
+        if output_option in ["script_text", "both"]:
+            script_path = os.path.join(temp_dir, f"script_{hash(input_text)}.txt")
+            with open(script_path, "w", encoding='utf-8') as f:
+                f.write(input_text)
+            script_file = script_path
+        
+        status_message = "Generation completed successfully!"
+        return audio_file, script_file, status_message
             
     except Exception as e:
-        return None, f"Error: {str(e)}"
-    
-    return None, None
+        return None, None, f"Error: {str(e)}"