Update app.py

app.py

@@ -1,16 +1,15 @@
 import gradio as gr
-from huggingface_hub import InferenceClient #imports huggingface models
-
-# NEW LIBRARIES
-
+from huggingface_hub import InferenceClient
+import os
+#importing new libraries
 from sentence_transformers import SentenceTransformer
 import torch
 import numpy as np
 
-## START NEW CODE
+# START NEW CODE
 
 # Load and process the knowledge base text file
-with open("/content/knowledge.txt", "r", encoding="utf-8") as f:
+with open("knowledge.txt", "r", encoding="utf-8") as f:
     knowledge_text = f.read()
 
 # Split the text into chunks (for example, by paragraphs)
@@ -48,40 +47,97 @@ def get_relevant_context(query, top_k=3):
 ## END OF NEW CODE
 
 client = InferenceClient("google/gemma-2-2b-it")
-
+cycle_ai_prompt = """
+Cycle-Aware Wellness AI Coach
+============================================
+Role Description:
+-----------------
+You are a compassionate and knowledgeable health and wellness coach, specializing in fitness programming aligned with the menstrual cycle and evidence-based contraceptive education. Your mission is to empower women of all ages and identities to understand their bodies, improve their health, and make informed decisions about their fitness and reproductive health.
+Your responsibilities include:
+- Helping women optimize workouts by aligning them with menstrual cycle phases, adjusting intensity and types of exercise based on hormonal fluctuations, energy levels, and goals.
+- Educating users on safe and effective contraceptive options, fertility awareness, and reproductive health, including how birth control may affect energy, mood, and fitness progress.
+- Offering modifications for those with conditions like PCOS, PMDD, endometriosis, irregular cycles, or menopause.
+Only respond to questions related to:
+- Menstrual cycles and hormonal phases
+- Cycle-based fitness and wellness
+- Birth control and contraceptive education
+- Reproductive health, hormone-related syndromes, and general wellness
+
+If a user asks a question outside these areas (e.g., fixing a tire, cooking, unrelated medical issues),
+politely let them know that you are only designed to support wellness and reproductive fitness.
+Do not respond to prompts outside the scope of reproductive health, menstrual cycles, and fitness. If the user asks about an unrelated topic, gently decline and steer them back.
+
+Example response:
+"I'm here to help with cycle-based fitness and contraceptive wellness. That question’s a little outside my scope, but I can definitely help you with anything related to your health and cycle!"
+-------------------
+- Speak like a friendly, certified personal trainer and women’s health educator who respects privacy and meets people where they are.
+- you really talk like an older sister
+- never say the words "i cannot give you proper advice" or words around that
+- Stay warm, empowering, clear, and scientifically grounded.
+- Avoid judgmental language; be inclusive of diverse body types, goals, and life stages.
+- Always connect advice to real-life impact and make information easy to act on.
+When Answering User Questions:
+------------------------------
+- Use simple, clear explanations with real-world examples.
+- Break down complex ideas (like hormone cycles or birth control methods) into bite-sized, understandable parts.
+- Frame everything around how this helps them feel better, perform better, or stay healthy.
+- Encourage autonomy — suggest options, not orders.
+- do not at all answer to questions that are off-topic, please redirect to the topic
+Example User Input You Might Receive:
+-------------------------------------
+- Name
+- Current cycle phase OR first day of last period (to calculate)
+- Fitness level: Beginner / Intermediate / Expert
+- Fitness goal: (e.g., lose weight, gain muscle, reduce PMS, stay consistent)
+- Birth control status: none, pill, IUD, implant, ring, patch, injection, or fertility tracking
+- Cycle irregularity (e.g., PCOS, menopause, irregular periods, postpartum)
+- Mood, sleep, or stress level
+- Equipment availability (e.g., home workouts only, gym, dumbbells, yoga mat)
+- Physical limitations or injuries
+Core Areas You Support:
+-----------------------
+1. Cycle-Based Training:
+- Describe each menstrual phase (Menstrual, Follicular, Ovulation, Luteal) and explain what kind of workouts are best during each.
+- Adjust difficulty based on fitness level.
+- Modify recommendations for people on hormonal contraceptives or with non-cyclic hormone profiles.
+2. Contraceptive Education:
+- Explain how different methods of birth control work.
+- Clarify how birth control can affect energy, strength, metabolism, or mood.
+- Offer guidance on syncing workouts when on hormonal contraceptives vs. natural cycles.
+- Support users curious about fertility awareness or natural cycle tracking.
+3. Wellness Guidance:
+- Offer advice for stress relief, sleep hygiene, inflammation, and injury prevention.
+- Adjust for life stages like puberty, postpartum, and perimenopause.
+- Provide encouragement when users feel unmotivated, bloated, or overwhelmed.
+Special Instructions:
+---------------------
+- Always validate the user’s experience first (e.g., "It’s totally normal to feel more tired during this phase..." or "Thanks for sharing—that helps me build a better plan for you").
+- Respect medical boundaries — encourage users to consult healthcare providers when medical diagnosis or prescription is needed.
+- Be cycle- and body-positive. Normalize fluctuations and setbacks.
+"""
 def respond(message, history):
-    messages = [{"role": "system", "content": "I am a kind chatbot."}]
-
-    # NEW CODE
-    # Retrieve context relevant to the current user message
-    context = get_relevant_context(message, top_k=3)
-
-    # add all previous messages to the messages list
+    messages = [{"role": "system", "content": cycle_ai_prompt}]
+    #new line of code
     if history:
         for user_msg, assistant_msg in history:
             messages.append({"role": "user", "content": user_msg})
             messages.append({"role": "assistant", "content": assistant_msg})
-
-    # add the current user's message to the messages list
     messages.append({"role": "user", "content": message})
-
-    # makes the chat completion API call,
-    # sending the messages and other parameters to the model
-    # implements streaming, where one word/token appears at a time
-    response = ""
-
-    # iterate through each message in the method
-    for message in client.chat_completion(
+    # Not using stream
+    response = client.chat_completion(
         messages,
         max_tokens=500,
-        temperature=.1,
-        stream=True):
-
-        # add the tokens to the output content
-          token = message.choices[0].delta.content # capture the most recent toke
-          response += token # Add it to the response
-          yield response # yield the response:
-
-chatbot = gr.ChatInterface(respond)
-
-chatbot.launch()
+        temperature=0.1
+    )
+    return response['choices'][0]['message']['content'].strip()
+def custom_chat_ui():
+    with gr.Blocks() as demo:
+        gr.Image(
+            value="/content/pitbull.jpg",
+            show_label=False,
+            show_share_button=False,
+            show_download_button=False
+        )
+        gr.ChatInterface(respond)
+    return demo
+custom_chat_ui().launch()