Upload 5 files

src/story_gen.py

@@ -4,12 +4,13 @@ from gtts import gTTS
 from langchain_groq import ChatGroq
 from langchain.messages import HumanMessage, SystemMessage
 import os
+from app_ui.utils.utils import app_config_col
 
 # Load variables from .env
 load_dotenv()
 
 # Access key
-groq_api_key = os.getenv("GROQ_API_KEY")
+groq_api_key = app_config_col.get("GROQ_API_KEY")
 
 chat = ChatGroq( 
     model="openai/gpt-oss-20b",
@@ -67,17 +68,20 @@ Write the story in STYLE style with:
 """
 
 
-def generate_story(story_plot,language):
+def generate_story(story_plot,config):
     messages=[
-        SystemMessage(content=sys_prompt.replace("LANGUAGE",language[0]).replace('LANG',language[1])),
+        SystemMessage(content=sys_prompt.replace("LANGUAGE",config.get('lang')[0]).replace('LANG',config.get('lang')[1])),
         HumanMessage(content=story_plot)
     ]
     response = chat.invoke(messages)
-    fileName=f"file_gtt_{uuid.uuid4().hex}.wav"
+
+
+    user_name=config.get('user')
+    fileName=f"{user_name}_{uuid.uuid4().hex}.wav"
 
     
 
-    tts = gTTS(text=response.content, lang=language[1])
+    tts = gTTS(text=response.content, lang=config.get('lang')[1])
     tts.save(fileName)
 
     return {