Update app.py

app.py

@@ -1,25 +1,47 @@
 import gradio as gr
-from datetime import datetime
 from huggingface_hub import InferenceClient
 import os
-# Set your Hugging Face token and model
-os.environ["HF_TOKEN"] = "your_huggingface_token_here"
-client = InferenceClient("google/gemma-2b-it")
-# System prompt for the assistant
+from sentence_transformers import SentenceTransformer
+import torch
+import numpy as np
+from datetime import datetime, timedelta
+
+# 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()]
+embedder = SentenceTransformer('all-MiniLM-L6-v2')
+chunk_embeddings = embedder.encode(chunks, convert_to_tensor=True)
+
+def get_relevant_context(query, top_k=3):
+    query_embedding = embedder.encode(query, convert_to_tensor=True)
+    query_embedding = query_embedding / query_embedding.norm()
+    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])
+    return context
+
+client = InferenceClient("google/gemma-2-2b-it")
+
 cycle_ai_prompt = """
 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.
@@ -27,114 +49,149 @@ 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, cooking, legal, or life advice.
 """
-# Global variables to persist across chats
-user_profile = {}
-chat_history = []
-# Calculate cycle phase
-def calculate_cycle_phase(start_date, end_date):
-    today = datetime.today().date()
-    start = datetime.strptime(start_date, "%Y-%m-%d").date()
-    end = datetime.strptime(end_date, "%Y-%m-%d").date()
-    cycle_length = 28  # Average cycle
-    days_since_start = (today - start).days
-    if days_since_start < 0:
-        return ":warning: Your entered period start date is in the future."
-    day_in_cycle = days_since_start % cycle_length
-    period_length = (end - start).days
-    if day_in_cycle <= period_length:
-        return "Menstrual phase :drop_of_blood:"
-    elif day_in_cycle <= 13:
-        return "Follicular phase :seedling:"
-    elif day_in_cycle <= 16:
-        return "Ovulation phase :dizzy:"
-    else:
-        return "Luteal phase :full_moon:"
-# Save user info
-def save_user_profile(name, age, level, period_start, period_end):
-    user_profile.update({
-        "name": name,
-        "age": age,
-        "level": level,
-        "period_start": period_start,
-        "period_end": period_end
-    })
-    phase = calculate_cycle_phase(period_start, period_end)
-    welcome = f":wave: Hi {name}! Based on your last period, you're currently in the **{phase}**.\n\nHow can I support your wellness today?"
-    return welcome
-# Main chatbot function
+
+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 = []
-    # Build conversation with system + chat history
+
     messages = [{"role": "system", "content": cycle_ai_prompt}]
-    # Append user profile to context as assistant message
-    if user_profile:
-        profile_intro = (
-            f"User Info:\n"
-            f"- Name: {user_profile.get('name')}\n"
-            f"- Age: {user_profile.get('age')}\n"
-            f"- Experience Level: {user_profile.get('level')}\n"
-            f"- Period Start: {user_profile.get('period_start')}\n"
-            f"- Period End: {user_profile.get('period_end')}\n"
-            f"- Current Phase: {calculate_cycle_phase(user_profile.get('period_start'), user_profile.get('period_end'))}"
-        )
-        messages.append({"role": "assistant", "content": profile_intro})
-    for turn in history:
-        if isinstance(turn, list) and len(turn) == 2:
-            messages.append({"role": "user", "content": turn[0]})
-            messages.append({"role": "assistant", "content": turn[1]})
+    for entry in history:
+        if isinstance(entry, dict):
+            messages.append(entry)
     messages.append({"role": "user", "content": message})
-    # Get response from model
-    try:
-        response = client.chat_completion(messages=messages, max_tokens=500, temperature=0.5)
-        assistant_reply = response["choices"][0]["message"]["content"].strip()
-    except Exception as e:
-        assistant_reply = f":warning: Something went wrong: {e}"
-    history.append((message, assistant_reply))
-    return history, ""
-# Gradio UI
-with gr.Blocks() as app:
-    gr.Markdown("# 🩷 Menstrual Cycle & Wellness Chatbot")
+
+    response = client.chat_completion(
+        messages,
+        max_tokens=500,
+        temperature=0.1
+    )
+
+    assistant_reply = response['choices'][0]['message']['content'].strip()
+    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);
+    }
+    """,
+    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:
+    name_state = gr.State("")
+    chat_history = gr.State([])
+
     with gr.Row():
         with gr.Column(scale=1):
-            gr.Markdown("### :woman_standing: Your Profile")
-            name = gr.Textbox(label="Name")
-            age = gr.Textbox(label="Age")
-            level = gr.Dropdown(["Beginner", "Intermediate", "Expert"], label="Experience Level")
-            period_start = gr.Textbox(label="Period Start Date (YYYY-MM-DD)")
-            period_end = gr.Textbox(label="Period End Date (YYYY-MM-DD)")
-            save_btn = gr.Button("Save Profile")
-            profile_output = gr.Markdown()
+            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="Cycle Chat").style(height=400)
-            user_msg = gr.Textbox(placeholder="Ask about your workouts, cycle, birth control, etc…")
-            send_btn = gr.Button("Send")
-    save_btn.click(
-        fn=save_user_profile,
-        inputs=[name, age, level, period_start, period_end],
-        outputs=profile_output
+            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]
     )
-    send_btn.click(
-        fn=respond,
-        inputs=[user_msg, chatbot],
-        outputs=[chatbot, user_msg]
+
+    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]
     )
-app.launch()
+
+if __name__ == '__main__':
+    demo.launch()