Update app.py

app.py

@@ -1,47 +1,25 @@
 import gradio as gr
+from datetime import datetime
 from huggingface_hub import InferenceClient
 import os
-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")
-
+# 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
 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:
+If the user asks something unrelated (e.g., fixing a car, meal prep, unrelated illnesses, tech support), immediately respond with the following message:
 "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.
@@ -49,170 +27,114 @@ 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.
-
-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!"
-
+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.
 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.
 """
-
-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."
-
+# 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 respond(message, history):
     if not isinstance(history, list):
         history = []
-
+    # Build conversation with system + chat history
     messages = [{"role": "system", "content": cycle_ai_prompt}]
-    for entry in history:
-        if isinstance(entry, dict):
-            messages.append(entry)
+    # 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]})
     messages.append({"role": "user", "content": message})
-
-    response = client.chat_completion(
-        messages,
-        max_tokens=500,
-        temperature=0.1
-    )
-
-    print("DEBUG RESPONSE:", response)  # <-- Add this to inspect
-
-    # Now adapt based on actual response
+    # Get response from model
     try:
-        assistant_reply = response['choices'][0]['message']['content'].strip()
+        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"⚠️ 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: #FFE4D1;
-        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;
-        margin-top: -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="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([])
-
+        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")
     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 →_")
-
+            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()
         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]
+            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
     )
-
-    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]
+    send_btn.click(
+        fn=respond,
+        inputs=[user_msg, chatbot],
+        outputs=[chatbot, user_msg]
     )
-
-if __name__ == '__main__':
-    demo.launch()
+app.launch()