Update ai_config.py

ai_config.py

@@ -1,3 +1,5 @@
+from io import BytesIO
+
 from langchain_openai import ChatOpenAI
 from openai import OpenAI
 import tiktoken
@@ -7,7 +9,8 @@ def n_of_questions():
     n_of_questions = 25
     return n_of_questions
 
-openai_api_key = os.environ.get("openai_api_key")
+#openai_api_key = os.environ.get("openai_api_key")
+openai_api_key = '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'
 
 model = "gpt-4o-mini"
 
@@ -22,20 +25,33 @@ def load_model(openai_api_key):
 # Initialize the OpenAI client with the API key
 client = OpenAI(api_key=openai_api_key)
 
+import os
+
 
-def convert_text_to_speech(text, output_file):
+def convert_text_to_speech(text, output):
     try:
         # Convert the final text to speech
         response = client.audio.speech.create(model="tts-1", voice="alloy", input=text)
 
-        with open(output_file, 'wb') as f:
+        if isinstance(output, BytesIO):
+            # If output is a BytesIO object, write directly to it
             for chunk in response.iter_bytes():
-                f.write(chunk)
+                output.write(chunk)
+        else:
+            # If output is a file path, open and write to it
+            with open(output, 'wb') as f:
+                for chunk in response.iter_bytes():
+                    f.write(chunk)
 
     except Exception as e:
         print(f"An error occurred: {e}")
+        # Fallback in case of error
         response = client.audio.speech.create(model="tts-1", voice="alloy", input='Here is my Report!')
 
-        with open(output_file, 'wb') as f:
+        if isinstance(output, BytesIO):
             for chunk in response.iter_bytes():
-                f.write(chunk)
+                output.write(chunk)
+        else:
+            with open(output, 'wb') as f:
+                for chunk in response.iter_bytes():
+                    f.write(chunk)