Upload 7 files

Dockerfile

@@ -1,23 +1,25 @@
-FROM python:3.13-slim
-
-WORKDIR /app
-
-
-# Install PortAudio + tools
-RUN apt-get update && apt-get install -y \
-    portaudio19-dev \
-    libportaudio2 \
-    libportaudiocpp0 \
-    && apt-get clean && rm -rf /var/lib/apt/lists/*
-
-COPY requirements.txt ./
-COPY src/ ./src/
-COPY app.py ./
-
-RUN pip3 install -r requirements.txt
-
-EXPOSE 8501
-
-#HEALTHCHECK CMD curl --fail http://localhost:8501/_stcore/health
-
-ENTRYPOINT ["streamlit", "run", "app.py"]
+FROM python:3.13-slim
+
+WORKDIR /app
+
+
+# Install PortAudio + tools
+RUN apt-get update && apt-get install -y \
+    portaudio19-dev \
+    libportaudio2 \
+    libportaudiocpp0 \
+    && apt-get clean && rm -rf /var/lib/apt/lists/*
+
+COPY requirements.txt ./
+COPY src/ ./src/
+COPY app.py ./
+COPY main.py ./
+COPY login.py ./
+
+RUN pip3 install -r requirements.txt
+
+EXPOSE 8501
+
+#HEALTHCHECK CMD curl --fail http://localhost:8501/_stcore/health
+
+ENTRYPOINT ["streamlit", "run", "login.py"]

app.py

@@ -1,27 +1,98 @@
 import streamlit as st
 from src.story_gen import generate_story
 from pathlib import Path
+import threading
+import time
+import os
 
-st.set_page_config(page_title="AI Story Generator", page_icon="📚", layout="centered")
+# ------------------- WAV CLEANUP CONFIG -------------------
+WAV_FOLDER = "."   # Change to your folder name
+DELETE_AFTER_SECONDS = 300
 
-st.title("📚 AI Story Generator")
-st.write("Enter a story plot and let AI turn it into a full story!")
 
-plot = st.text_area("Enter your story plot:", height=150)
+def delete_wav_files():
+    """Delete .wav files older than 1 hour."""
+    if not os.path.exists(WAV_FOLDER):
+        return
 
-generate = st.button("Generate Story")
+    current_time = time.time()
 
-if generate:
-    if not plot.strip():
-        st.warning("Please enter a story plot.")
-    else:
-        with st.spinner("Generating story... please wait ⏳"):
-            response = generate_story(story_plot=plot)
-            story = response.get('story')
-            fileName=response.get('fileName')
+    for filename in os.listdir(WAV_FOLDER):
+        if filename.endswith(".wav"):
+            file_path = os.path.join(WAV_FOLDER, filename)
 
-        st.subheader("✨ Generated Story")
-        #st.write(story)
-        f=Path(fileName)
-        audio_bytes = f.read_bytes()
-        st.audio(audio_bytes, format="audio/wav")
+            try:
+                file_age = current_time - os.path.getmtime(file_path)
+
+                if file_age > DELETE_AFTER_SECONDS:
+                    os.remove(file_path)
+                    print(f"[CLEANER] Deleted (older than 1 hr): {file_path}")
+
+            except Exception as e:
+                print(f"[CLEANER] Error deleting {file_path}: {e}")
+
+def background_cleaner():
+    """Runs every 1 minute."""
+    while True:
+        delete_wav_files()
+        time.sleep(60)
+
+
+def load_app():
+    # Start background cleaner only once
+    if "bg_started" not in st.session_state:
+        thread = threading.Thread(target=background_cleaner, daemon=True)
+        thread.start()
+        st.session_state.bg_started = True
+        print("[CLEANER] Background WAV cleaner started.")
+
+
+    # ------------------- STREAMLIT APP -------------------
+
+    st.set_page_config(page_title="AI Story Generator", page_icon="📚", layout="centered")
+
+    st.title("📚 AI Story Generator")
+    st.write("Enter a story plot and let AI turn it into a full story!")
+
+    # 🔽 Language dropdown
+    language = st.selectbox(
+        "Select language for audio:",
+        [
+            ("English", "en"),
+            ("Hindi", "hi"),
+            ("Tamil", "ta"),
+            ("Telugu", "te"),
+            ("Bengali", "bn"),
+            ("Gujarati", "gu"),
+            ("Kannada", "kn"),
+            ("Malayalam", "ml")
+        ],
+        format_func=lambda x: x[0]
+    )
+
+    plot = st.text_area("Enter your story plot:", height=150)
+
+    display_story=st.checkbox(label="Display Story")
+
+    generate = st.button("Generate Story")
+
+    if generate:
+        if not plot.strip():
+            st.warning("Please enter a story plot.")
+        else:
+            with st.spinner("Generating story... please wait ⏳"):
+
+                # Pass language to your story generator
+                response = generate_story(story_plot=plot, language=language)
+
+                story = response.get('story')
+                fileName = response.get('fileName')
+
+            st.subheader("✨ Generated Story")
+
+            f = Path(fileName)
+            audio_bytes = f.read_bytes()
+            st.audio(audio_bytes, format="audio/wav")
+
+            if display_story:
+                st.write(story)

login.py

@@ -0,0 +1,100 @@
+import streamlit as st
+from hashlib import sha256
+from app import load_app
+import os
+import json
+from dotenv import load_dotenv
+
+load_dotenv()
+
+# -----------------------
+# Mock user database (replace with DB in production)
+# -----------------------
+
+USER_CONFIG = json.loads(os.getenv('USER_CONFIG'))
+
+print("********************************************")
+print(type(USER_CONFIG))
+print(USER_CONFIG)
+# -----------------------
+# Authentication Utilities
+# -----------------------
+def hash_password(password: str) -> str:
+    return sha256(password.encode()).hexdigest()
+
+def is_authenticated(username: str, password: str) -> bool:
+    if username in USER_CONFIG:
+        return USER_CONFIG[username] == password
+    return False
+
+def login_page():
+    st.markdown(
+        """
+        <h2 style="text-align:center;margin-bottom:30px;">🔐 Secure Login</h2>
+        """,
+        unsafe_allow_html=True,
+    )
+
+    with st.form("login_form"):
+        username = st.text_input("Username", placeholder="Enter username")
+        password = st.text_input("Password", type="password", placeholder="Enter password")
+        submit = st.form_submit_button("Login")
+
+    if submit:
+        if is_authenticated(username, password):
+            st.session_state["auth"] = True
+            st.session_state["user"] = username
+            st.success("Login successful! Redirecting...")
+            st.rerun()
+        else:
+            st.error("Invalid username or password")
+
+
+def logout_button():
+    # --- TOP CENTER WARNING ---
+    st.markdown(
+        """
+        <div style="text-align:center; margin-top:-20px;">
+            <p style="color:#5cb85c; font-size:20px; font-weight:bold;">
+                Logged in as: <span style="color:black;">{}</span>
+            </p>
+        </div>
+        """.format(st.session_state['user']),
+        unsafe_allow_html=True
+    )
+
+    # --- TOP RIGHT LOGOUT BUTTON ---
+    logout_col = st.columns([8, 2])  # Push button to the right
+    with logout_col[1]:
+        if st.button("Logout", key="logout_btn"):
+            st.session_state["auth"] = False
+            st.session_state["user"] = None
+            st.rerun()
+
+
+# -----------------------
+# Main App (Protected)
+# -----------------------
+def main_app():
+    logout_button()
+    load_app()
+
+
+# -----------------------
+# Streamlit Entry Point
+# -----------------------
+def main():
+    # Initialize session state
+    if "auth" not in st.session_state:
+        st.session_state["auth"] = False
+        st.session_state["user"] = None
+
+    # Render login or app
+    if not st.session_state["auth"]:
+        login_page()
+    else:
+        main_app()
+
+
+if __name__ == "__main__":
+    main()

main.py

@@ -0,0 +1,6 @@
+def main():
+    print("Hello from story-app!")
+
+
+if __name__ == "__main__":
+    main()

src/story_gen.py

@@ -20,7 +20,8 @@ chat = ChatGroq(
 
 
 sys_prompt="""
-        You are a professional story writer who creates engaging, vivid, emotionally rich stories based on a user's plot or idea.
+You are a professional story writer who creates engaging, vivid, emotionally rich stories 
+based on a user's plot or idea.
 
 Your goal:
 - Expand the user's plot into a complete story.
@@ -52,6 +53,9 @@ Your output:
 
 When ready, ask the user:
 “Please share your plot or idea for the story.”
+
+Note: Write the story on LANGUAGE only and this story will be given to gtts using LANG for audio 
+generation so consider this while creating story
 """
 
 simple_prompt="""
@@ -63,15 +67,17 @@ Write the story in STYLE style with:
 """
 
 
-def generate_story(story_plot):
+def generate_story(story_plot,language):
     messages=[
-        SystemMessage(content=sys_prompt),
+        SystemMessage(content=sys_prompt.replace("LANGUAGE",language[0]).replace('LANG',language[1])),
         HumanMessage(content=story_plot)
     ]
     response = chat.invoke(messages)
     fileName=f"file_gtt_{uuid.uuid4().hex}.wav"
 
-    tts = gTTS(text=response.content, lang='en')
+    
+
+    tts = gTTS(text=response.content, lang=language[1])
     tts.save(fileName)
 
     return {
@@ -80,5 +86,5 @@ def generate_story(story_plot):
     }
 
 
-# res=generate_story("write a story on taling lion and river")
+# res=generate_story(story_plot="write a story on lion and river",language="assamese")
 # print(res.get('story'))