Update app.py

app.py

@@ -2,9 +2,10 @@ import gradio as gr
 from datetime import datetime
 from huggingface_hub import InferenceClient
 import os
-os.environ["HF_TOKEN"] = "your_huggingface_token_here"  # Replace with your HF token
-client = InferenceClient("google/gemma-2b-it")  # Use your preferred model
-# Prompt template
+# 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)
 ==========================================================
@@ -44,78 +45,96 @@ 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.
 """
-# Track user profile in memory
-user_profile = {"name": "", "age": "", "level": "", "period_start": "", "period_end": ""}
-# Cycle phase calculator
+# 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
+    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
-    if day_in_cycle <= (end - start).days:
+    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:"
-    elif day_in_cycle <= 28:
-        return "Luteal phase :full_moon:"
     else:
-        return "Unknown phase :question:"
-# Save profile and return welcome
-def save_user_profile(name, age, level, period_start, period_end, history):
-    user_profile["name"] = name
-    user_profile["age"] = age
-    user_profile["level"] = level
-    user_profile["period_start"] = period_start
-    user_profile["period_end"] = period_end
+        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: Hello {name}! Based on your last period, you're currently in the {phase}."
-    history.append({"role": "assistant", "content": welcome})
-    return history, welcome
-# Respond to user input
+    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 = []
-    messages = [{"role": "system", "content": cycle_ai_prompt}] + 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]})
     messages.append({"role": "user", "content": message})
-    response = client.chat_completion(messages, max_tokens=500, temperature=0.2)
+    # Get response from model
     try:
-        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:
-        reply = f":warning: Error generating response: {e}"
-    history.append({"role": "user", "content": message})
-    history.append({"role": "assistant", "content": reply})
-    return history, history, ""
-# UI
+        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 Chatbot")
+    gr.Markdown("# 🩷 Menstrual Cycle & Wellness Chatbot")
     with gr.Row():
         with gr.Column(scale=1):
-            gr.Markdown("### :woman_standing: Your Info")
-            name_input = gr.Textbox(label="Name")
-            age_input = gr.Textbox(label="Age")
-            level_input = gr.Dropdown(choices=["Beginner", "Intermediate", "Expert"], label="Experience Level")
-            period_start_input = gr.Textbox(label="Period Start (YYYY-MM-DD)")
-            period_end_input = gr.Textbox(label="Period End (YYYY-MM-DD)")
-            set_button = gr.Button("Save Profile")
+            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="Cycle Chat", type="messages", value=[])
-            user_input = gr.Textbox(placeholder="Ask me anything about your cycle…")
-            submit_btn = gr.Button("Send")
-    set_button.click(
+            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_input, age_input, level_input, period_start_input, period_end_input, chatbot],
-        outputs=[chatbot, profile_output]
+        inputs=[name, age, level, period_start, period_end],
+        outputs=profile_output
     )
-    submit_btn.click(
+    send_btn.click(
         fn=respond,
-        inputs=[user_input, chatbot],
-        outputs=[chatbot, chatbot, user_input]
+        inputs=[user_msg, chatbot],
+        outputs=[chatbot, user_msg]
     )
 app.launch()