new code

app.py

@@ -1,10 +1,12 @@
 import gradio as gr
 from huggingface_hub import InferenceClient
+import os
 from sentence_transformers import SentenceTransformer
 import torch
 import numpy as np
+from datetime import datetime, timedelta
 
-# Load and process the knowledge base
+# Load knowledge base
 with open("knowledge.txt", "r", encoding="utf-8") as f:
     knowledge_text = f.read()
 chunks = [chunk.strip() for chunk in knowledge_text.split("\n\n") if chunk.strip()]
@@ -17,7 +19,7 @@ def get_relevant_context(query, top_k=3):
     norm_chunk_embeddings = chunk_embeddings / chunk_embeddings.norm(dim=1, keepdim=True)
     similarities = torch.matmul(norm_chunk_embeddings, query_embedding)
     top_k_indices = torch.topk(similarities, k=top_k).indices.cpu().numpy()
-    context = "\n\n".join([chunks[i] for i in top_k_indices])
+    context = "\n\n".join([chunks[i]] for i in top_k_indices)
     return context
 
 client = InferenceClient("google/gemma-2-2b-it")
@@ -28,14 +30,18 @@ Cycle-Aware Wellness AI Coach (Strict Enforcement Version)
 Mission:
 --------
 You are a compassionate and knowledgeable wellness coach who specializes in fitness aligned with the menstrual cycle and evidence-based contraceptive education. Your mission is to empower women to understand their bodies, support their fitness and reproductive health, and make informed, cycle-aware choices.
+
 You may *only* respond to questions related to:
 - Menstrual cycles and hormonal phases
 - Cycle-based fitness and wellness programming
 - Contraceptive methods and reproductive health education
 - Hormonal syndromes or life stages (PCOS, PMDD, menopause, irregular cycles, etc.)
-If the user asks something unrelated (e.g., fixing a car, meal prep, unrelated illnesses, tech support), immediately respond with the following message:
+
+If the user asks something unrelated (e.g., fixing a car, meal prep, unrelated illnesses, tech support), immediately respond with:
 "I'm here to help with cycle-based fitness and contraceptive wellness. That question’s outside my scope, but I’d love to support you with anything related to your body, cycle, or health goals!"
+
 You must *never* attempt to answer off-topic requests, even if the user insists or rephrases. Always redirect the conversation back to wellness, hormones, fitness, or reproductive health.
+
 Style and Voice Guidelines:
 ---------------------------
 - Speak like a friendly, knowledgeable older sister who’s also a certified personal trainer and women’s health educator.
@@ -43,59 +49,169 @@ Style and Voice Guidelines:
 - Validate the user’s experience before offering guidance (e.g., “That makes total sense—your energy might be shifting in this phase”).
 - Encourage autonomy by offering options, not orders.
 - Connect advice to real-life impact: how the cycle affects energy, mood, and performance.
-Core Support Areas (Scope You *Do* Cover):
-------------------------------------------
-1. Cycle-Based Training:
-- Explain hormonal phases (Menstrual, Follicular, Ovulation, Luteal).
-- Suggest workouts aligned to the user’s cycle and fitness level.
-- Adjust for users on birth control, with hormone conditions, or in life stages like menopause or postpartum.
-2. Contraceptive Education:
-- Educate on birth control methods (pill, IUD, implant, etc.).
-- Explain how contraception affects workouts, mood, metabolism.
-- Support users interested in natural tracking or switching methods.
-3. Wellness Support:
-- Help with stress, motivation, sleep, inflammation, and hormone-related fatigue or soreness.
-- Normalize fluctuations and setbacks.
-- Offer encouragement when users feel overwhelmed, bloated, or inconsistent.
+
+When the user asks about workouts, always ask a follow-up:
+"What kind of equipment or space do you have access to—like a gym, home weights, or just bodyweight? I’ll tailor a workout for you based on that!"
+
 Final Boundary Rule:
 --------------------
-🔒 Strictly decline all unrelated questions. Your only purpose is cycle-aware fitness and reproductive wellness coaching. Do not give general medical, tech, legal, or life advice. You can still give advice relating to helping symptoms like home remedies. No actual medical advice, but recipes are fine.
+ Strictly decline all unrelated questions. Your only purpose is cycle-aware fitness and reproductive wellness coaching. Do not give general medical, tech, cooking, legal, or life advice.
 """
 
+def determine_cycle_phase(start_date_str):
+    try:
+        start_date = datetime.strptime(start_date_str, "%Y-%m-%d")
+        days_since = (datetime.now() - start_date).days % 28
+        if days_since < 5:
+            return "Menstrual Phase", "💗 Time to rest and recover. Gentle movement like stretching or walking is great."
+        elif days_since < 13:
+            return "Follicular Phase", "💪 Your energy’s building—go for strength training or cardio!"
+        elif days_since < 16:
+            return "Ovulation Phase", "🔥 Peak power! Try high-intensity workouts or social activities."
+        else:
+            return "Luteal Phase", "�� Wind down. Opt for lighter training, yoga, or bodyweight exercises."
+    except:
+        return "Unknown Phase", "Couldn't parse the date. Please use YYYY-MM-DD."
+
 def respond(message, history):
+    if not isinstance(history, list):
+        history = []
+
     messages = [{"role": "system", "content": cycle_ai_prompt}]
-    if history:
-        for user_msg, assistant_msg in history:
-            messages.append({"role": "user", "content": user_msg})
-            messages.append({"role": "assistant", "content": assistant_msg})
+    for entry in history:
+        if isinstance(entry, dict):
+            messages.append(entry)
     messages.append({"role": "user", "content": message})
+
     response = client.chat_completion(
         messages,
         max_tokens=500,
         temperature=0.1
     )
-    return response['choices'][0]['message']['content'].strip()
-
-def custom_chat_ui():
-    with gr.Blocks(
-        theme=gr.themes.Soft(
-            primary_hue="pink",
-            secondary_hue="orange",
-            neutral_hue="yellow",
-            spacing_size="lg",
-            radius_size="lg",  # ← FIXED: was "xl"
-            font=[gr.themes.GoogleFont("Quicksand"), "sans-serif"],
-            font_mono=[gr.themes.GoogleFont("IBM Plex Mono"), "monospace"]
-        )
-    ) as demo:
-        gr.Image(
-            show_label=False,
-            show_share_button=False,
-            show_download_button=False,
-            container=False,
-            height=200
-        )
-        gr.ChatInterface(respond, title="🌸 Cycle-Aware Wellness Coach 🌞")
-    return demo
-
-custom_chat_ui().launch()
+
+    print("DEBUG RESPONSE:", response)  # <-- Add this to inspect
+
+    # Now adapt based on actual response
+    try:
+        assistant_reply = response['choices'][0]['message']['content'].strip()
+    except Exception as e:
+        assistant_reply = f"⚠️ There was an error processing the response: {e}"
+
+    new_history = history + [
+        {"role": "user", "content": message},
+        {"role": "assistant", "content": assistant_reply}
+    ]
+    return new_history, new_history, ""
+
+def update_chatbot(user_message, history):
+    return respond(user_message, history)
+
+def set_user_info(name, age, level, period_start_date, period_end_date):
+    phase, tip = determine_cycle_phase(period_start_date)
+    greeting = f"Hi {name}! I'm here to help you with cycle-aware fitness and wellness.\n\nYou’re {age} years old, training at a {level.lower()} level, and your last period started on {period_start_date} and ended on {period_end_date}.\n\nRight now, you’re likely in your **{phase}**. {tip} 💞\n\nAsk me anything about your body, cycle, or contraceptive health!"
+    return name, [{"role": "assistant", "content": greeting}]
+
+def button_click(question, history):
+    new_history, updated_history, _ = respond(question, history)
+    return new_history, updated_history
+
+with gr.Blocks(
+    css="""
+    .gradio-container {
+        background: linear-gradient(135deg, #F6D365 0%, #FDA085 50%, #FF8FA0 100%);
+        font-family: 'Quicksand', sans-serif;
+    }
+    .message.user {
+        background-color: lightpink;
+        border-radius: 20px;
+        padding: 10px;
+        margin: 5px;
+        max-width: 75%;
+        align-self: flex-end;
+    }
+    .message.bot {
+        background-color: lightcoral;
+        border-radius: 20px;
+        padding: 10px;
+        margin: 5px;
+        max-width: 75%;
+        align-self: flex-start;
+    }
+    .chat-interface {
+        background-color: peachpuff;
+        border-radius: 12px;
+        padding: 15px;
+        box-shadow: 0 4px 10px rgba(0, 0, 0, 0.1);
+    }
+
+    #banner-image {
+        background-color: transparent;
+        margin-bottom: -100px;
+
+    }
+    """,
+    theme=gr.themes.Soft(
+        primary_hue="pink",
+        secondary_hue="orange",
+        neutral_hue="yellow",
+        spacing_size="lg",
+        radius_size="lg",
+        font=[gr.themes.GoogleFont("Quicksand"), "sans-serif"],
+        font_mono=[gr.themes.GoogleFont("IBM Plex Mono"), "monospace"]
+    )
+) as demo:
+    gr.Image(
+        value="/content/Untitled design.png",
+        show_label=False,
+        show_share_button = False,
+        show_download_button = False,
+        elem_id="banner-image")
+
+    name_state = gr.State("")
+    chat_history = gr.State([])
+
+    with gr.Row():
+        with gr.Column(scale=1):
+            gr.Markdown("### Tell me about yourself")
+            name_input = gr.Textbox(label="Name", placeholder="Your name…")
+            age_input = gr.Textbox(label="Age", placeholder="Your age…")
+            level_input = gr.Dropdown(choices=["Beginner", "Intermediate", "Expert"], label="Training Level")
+            period_start_input = gr.Textbox(label="Last Period Start Date", placeholder="YYYY-MM-DD")
+            period_end_input = gr.Textbox(label="Last Period End Date", placeholder="YYYY-MM-DD")
+            set_btn = gr.Button("Set Info")
+            gr.Markdown("_After the greeting appears, start chatting →_")
+
+        with gr.Column(scale=2):
+            chatbot = gr.Chatbot(label="Chat", type="messages")
+            user_text = gr.Textbox(placeholder="Ask me something…", label="")
+            with gr.Row():
+                workout_btn = gr.Button("Prompt: Workouts")
+                contraceptive_btn = gr.Button("Prompt: Contraceptives")
+
+    set_btn.click(
+        fn=set_user_info,
+        inputs=[name_input, age_input, level_input, period_start_input, period_end_input],
+        outputs=[name_state, chatbot],
+        show_progress=False,
+    )
+
+    user_text.submit(
+        fn=update_chatbot,
+        inputs=[user_text, chat_history],
+        outputs=[chatbot, chat_history, user_text]
+    )
+
+    workout_btn.click(
+        fn=lambda history: button_click("What workout routine would you recommend based on my cycle phase?", history),
+        inputs=[chat_history],
+        outputs=[chatbot, chat_history]
+    )
+
+    contraceptive_btn.click(
+        fn=lambda history: button_click("Can you explain the different contraceptive options and their benefits?", history),
+        inputs=[chat_history],
+        outputs=[chatbot, chat_history]
+    )
+
+if __name__ == '__main__':
+    demo.launch()