Update app.py

app.py

@@ -1,14 +1,12 @@
-import random
+ import random
 import streamlit as st
+import streamlit.components.v1 as components
 from sentence_transformers import SentenceTransformer, util
 from ensemble import get_ensembler
-from quiz_engine import run_quiz_step, generate_tarot_card_summary, generate_reflection_and_prompt
+from quiz_engine import run_quiz_step, generate_tarot_card_summary, generate_reflection_and_prompt, QUESTION_BANK
 from vibe_mapper import get_emojis_from_emotions
-import streamlit.components.v1 as components
-from quiz_engine import QUESTION_BANK
-
 
-# ✅ Initialize session state keys
+# Initialize session state
 if "quiz_started" not in st.session_state:
     st.session_state.quiz_started = False
 if "question_index" not in st.session_state:
@@ -20,7 +18,7 @@ if "reflections" not in st.session_state:
 if "show_final" not in st.session_state:
     st.session_state.show_final = False
 
-# 🎨 UI Layout with background and typing style
+# 🎨 Styling
 st.markdown("""
     <style>
     body {
@@ -42,11 +40,11 @@ st.markdown("""
     </style>
 """, unsafe_allow_html=True)
 
-# Optional background music
+# 🎵 Background Music
 if not st.session_state.quiz_started:
     st.audio("https://www.bensound.com/bensound-music/bensound-sunny.mp3", autoplay=True)
 
-# Show typing animation only BEFORE quiz starts
+# 💬 Typing Intro
 if not st.session_state.quiz_started:
     components.html("""
     <div style="text-align: center; font-size: 2em; font-family: 'Courier New', monospace; color: #d63384;">
@@ -83,22 +81,19 @@ if not st.session_state.quiz_started:
     </script>
     """, height=80)
 
-# 🚦 Quiz Flow Trigger
+# 🧠 Start Quiz Button
 if not st.session_state.quiz_started:
     if st.button("Let's Begin 💖", type="primary"):
         st.session_state.quiz_started = True
         st.rerun()
     st.info("🎼 This quiz is designed to gently uncover your emotional love style.")
 
+# 🧩 Quiz in Progress
 else:
-    # Typing animation for questions
-    if st.session_state.question_index < len(st.session_state.answers):
-        st.markdown(f"<div class='question-text'>{st.session_state.answers[st.session_state.question_index]['question']}</div>", unsafe_allow_html=True)
-    else:
-        run_quiz_step()
-
     if st.session_state.question_index < len(QUESTION_BANK):
         current_question = QUESTION_BANK[st.session_state.question_index]
+
+        # ✨ JS Typing Animation
         components.html(f"""
         <div style="text-align: center; font-size: 1.4em; font-family: 'Courier New', monospace; color: #ff75a0;">
         <span id="question-typing"></span>
@@ -118,15 +113,41 @@ else:
         </script>
         """, height=80)
 
-        st.markdown("## 💌 Final Reflection")
-        ensembler = get_ensembler()
+        # 💬 Input + Emotion Detection
+        user_input = st.text_input("Your response", key=f"q_{st.session_state.question_index}")
+
+        if user_input and "answered" not in st.session_state:
+            ensembler = get_ensembler()
+            tone = ensembler.predict_emotion(user_input)
+            st.session_state.answers.append({
+                "question": current_question,
+                "response": user_input,
+                "tone": tone
+            })
+            st.session_state.question_index += 1
+            st.session_state.answered = True
+            st.rerun()
+
+        if "answered" in st.session_state:
+            del st.session_state.answered
+
+    # ✅ End of Quiz: Show Final Results on Button Press
+    elif len(st.session_state.answers) >= len(QUESTION_BANK):
+        if not st.session_state.show_final:
+            if st.button("Reveal My Final Reflection 💖"):
+                st.session_state.show_final = True
+                st.rerun()
+        else:
+            ensembler = get_ensembler()
+            st.markdown("## 💌 Final Reflection")
+
+            all_text = " ".join([entry["response"] for entry in st.session_state.answers])
+            result = ensembler.predict_emotion(all_text)
 
-        all_text = " ".join([entry["response"] for entry in st.session_state.answers])
-        result = ensembler.predict_emotion(all_text)
+            st.subheader("💫 Your Emotional Reflection")
+            for label, score in result["top_emotions"]:
+                st.write(f"**{label.capitalize()}**: {score:.2f}")
 
-        st.subheader("💫 Your Emotional Reflection")
-        for label, score in result["top_emotions"]:
-            st.write(f"**{label.capitalize()}**: {score:.2f}")
             st.markdown("**Vibe Tags:** " + ", ".join(result["vibe_tags"]))
             st.success("🍃 Vibe Tags: " + ", ".join(result["vibe_tags"]))
 
@@ -140,7 +161,6 @@ else:
             else:
                 st.success("All clear — nothing spicy detected.")
 
-            # 🧠 Log All Q&A Entries
             st.markdown("---")
             st.subheader("📜 Your Answers Log")
             for idx, entry in enumerate(st.session_state.answers):
@@ -149,7 +169,6 @@ else:
                 if "reflection" in entry:
                     st.markdown(f"*Reflection:* {entry['reflection']}")
 
-            # 🔮 Generate Tarot Card Summary
             tarot_summary = generate_tarot_card_summary(st.session_state.reflections)
             st.markdown("---")
             st.subheader("🔮 Your Romantic Archetype")