Update app.py

app.py

@@ -1,11 +1,29 @@
 import gradio as gr
-from datetime import datetime
 from huggingface_hub import InferenceClient
 import os
-# Replace with your actual Hugging Face token
-os.environ["HF_TOKEN"] = "your_huggingface_token_here"
-client = InferenceClient("google/gemma-2b-it")
-# Prompt
+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)
 ==========================================================
@@ -17,7 +35,7 @@ You may *only* respond to questions related to:
 - 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:
@@ -27,92 +45,166 @@ 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.
+ 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.
 """
-# Determine the current 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
-    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
-    if day_in_cycle <= (end - start).days:
-        return "Menstrual phase :drop_of_blood:"
-    elif day_in_cycle <= 13:
-        return "Follicular phase :seedling:"
-    elif day_in_cycle <= 16:
-        return "Ovulation phase :dizzy:"
-    elif day_in_cycle <= 28:
-        return "Luteal phase :full_moon:"
-    else:
-        return "Unknown phase :question:"
-# Profile dictionary
-user_profile = {"name": "", "age": "", "level": "", "period_start": "", "period_end": ""}
-# Update profile 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)
-    return f":wave: Hello {name}! Based on your last period, you're currently in the **{phase}**."
-# Chat function
-def respond(message, chat_history):
-    if not isinstance(chat_history, list):
-        chat_history = []
-    # Construct prompt
+
+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}]
-    for turn in chat_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})
+
+    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
     try:
-        result = client.chat_completion(messages=messages, temperature=0.2, max_tokens=500)
-        reply = result["choices"][0]["message"]["content"].strip()
+        assistant_reply = response['choices'][0]['message']['content'].strip()
     except Exception as e:
-        reply = f":warning: There was an error: {str(e)}"
-    chat_history.append([message, reply])
-    return chat_history, chat_history
-# Interface
-with gr.Blocks() as app:
-    gr.Markdown("# 🩷 Menstrual Cycle Chatbot")
+        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([])
+
     with gr.Row():
         with gr.Column(scale=1):
-            gr.Markdown("### :woman_standing: Your Info")
-            name = gr.Textbox(label="Name")
-            age = gr.Textbox(label="Age")
-            level = gr.Dropdown(choices=["Beginner", "Intermediate", "Expert"], label="Experience Level")
-            start_date = gr.Textbox(label="Period Start Date (YYYY-MM-DD)")
-            end_date = gr.Textbox(label="Period End Date (YYYY-MM-DD)")
-            save_btn = gr.Button("Save Profile")
-            profile_out = 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 Support Chat", value=[], height=400)
-            user_msg = gr.Textbox(placeholder="Ask me anything about your cycle…", show_label=False)
-            send_btn = gr.Button("Send")
-    save_btn.click(save_user_profile, inputs=[name, age, level, start_date, end_date], outputs=profile_out)
-    send_btn.click(respond, inputs=[user_msg, chatbot], outputs=[chatbot, chatbot])
-app.launch()
+            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()