Update app.py

app.py

@@ -7,6 +7,17 @@ from fpdf import FPDF
 from gtts import gTTS
 import gradio as gr
 
+firebase_config = {
+    "apiKey": os.getenv("FIREBASE_API_KEY"),
+    "authDomain": os.getenv("FIREBASE_AUTH_DOMAIN"),
+    "databaseURL": os.getenv("FIREBASE_DATABASE_URL"),
+    "projectId": os.getenv("FIREBASE_PROJECT_ID"),
+    "storageBucket": os.getenv("FIREBASE_STORAGE_BUCKET"),
+    "messagingSenderId": os.getenv("FIREBASE_MESSAGING_SENDER_ID"),
+    "appId": os.getenv("FIREBASE_APP_ID")
+}
+
+
 from langchain.prompts import ChatPromptTemplate, SystemMessagePromptTemplate, HumanMessagePromptTemplate
 from langchain.memory import ConversationBufferMemory
 from llama_index.core import SimpleDirectoryReader, VectorStoreIndex
@@ -81,8 +92,13 @@ def save_story_to_file(character, story_text, age_group="kids", tone="", theme="
         "theme": theme,
         "timestamp": datetime.now().isoformat()
     }
-    with open("story_logs.json", "a", encoding="utf-8") as f:
-        f.write(json.dumps(story_data) + "\n")
+
+    if session["token"]:
+        db.child("stories").push({**story_data, "email": session["email"]})
+    else:
+        with open("story_logs.json", "a", encoding="utf-8") as f:
+            f.write(json.dumps(story_data) + "\n")
+
 
 # ---------------- Kids Prompt (ChatPromptTemplate) ----------------
 kids_prompt = ChatPromptTemplate.from_messages([
@@ -444,5 +460,19 @@ with gr.Blocks(title="📚 Children's Story Assistant") as app:
     refresh_btn = gr.Button("🔄 Load Stories")
     refresh_btn.click(fn=refresh_dropdown, outputs=story_selector)
     story_selector.change(fn=show_story_by_title, inputs=story_selector, outputs=story_output)
+    
+    with gr.Tab("🔐 Login"):
+       email_input = gr.Textbox(label="Email")
+       password_input = gr.Textbox(label="Password", type="password")
+       login_btn = gr.Button("Login")
+       login_msg = gr.Textbox(label="Status")
+       login_btn.click(login, inputs=[email_input, password_input], outputs=login_msg)
+
+    with gr.Tab("📝 Register"):
+       email_input2 = gr.Textbox(label="Email")
+       password_input2 = gr.Textbox(label="Password", type="password")
+       register_btn = gr.Button("Register")
+       register_msg = gr.Textbox(label="Status")
+       register_btn.click(register, inputs=[email_input2, password_input2], outputs=register_msg)
 
 app.launch()