refactored code

Dockerfile

@@ -0,0 +1,15 @@
+FROM python:3.8-slim
+
+WORKDIR /app
+
+COPY requirements.txt .
+RUN pip install --no-cache-dir -r requirements.txt
+
+COPY . .
+
+ENV PYTHONPATH="/app"
+ENV PORT=7860
+
+EXPOSE 7860
+
+CMD ["python", "app.py"] 

Dockerfile.rasa

@@ -0,0 +1,19 @@
+FROM python:3.8-slim
+
+WORKDIR /app
+
+RUN apt-get update && apt-get install -y curl && rm -rf /var/lib/apt/lists/*
+
+COPY requirements.txt .
+RUN pip install --no-cache-dir -r requirements.txt
+
+COPY ./rasa_app /app/rasa_app
+
+ENV PYTHONPATH="/app"
+
+EXPOSE 5005
+
+HEALTHCHECK --interval=30s --timeout=10s --start-period=15s --retries=3 \
+  CMD curl -f http://localhost:5005/ || exit 1
+
+CMD ["python", "-m", "rasa", "run", "--enable-api", "--cors", "*"] 

README.md

@@ -1,10 +1,14 @@
 ---
-title: IHCI Diet
-emoji: 📉
+title: Diet Planning Chatbot
+emoji: 🥗
 colorFrom: green
 colorTo: blue
 sdk: docker
+sdk_version: "3.0.0"
+app_file: app.py
 pinned: false
 ---
 
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+# Diet Planning Chatbot
+
+A personalized diet and meal planning assistant that creates custom meal plans based on your preferences, restrictions, and health goals. 

app.py

@@ -0,0 +1,506 @@
+import gradio as gr
+import cohere
+import os
+import re
+from dotenv import load_dotenv
+
+load_dotenv()
+COHERE_API_KEY = os.getenv("COHERE_API_KEY")
+co = cohere.Client(COHERE_API_KEY)
+
+custom_css = """
+.gradio-container {
+    font-family: 'Inter', -apple-system, BlinkMacSystemFont, sans-serif;
+}
+.app-header {
+    text-align: center;
+    margin-bottom: 20px;
+}
+.app-title {
+    background: linear-gradient(90deg, #2e7d32, #1976d2);
+    -webkit-background-clip: text;
+    -webkit-text-fill-color: transparent;
+    font-weight: 700;
+    font-size: 2.5em;
+    margin-bottom: 8px;
+}
+.markdown-text p {
+    margin-bottom: 12px;
+    line-height: 1.5;
+}
+.diet-header {
+    font-size: 1.5em;
+    margin-top: 24px;
+    margin-bottom: 16px;
+    font-weight: 600;
+    color: #2e7d32;
+    border-bottom: 1px solid rgba(0, 0, 0, 0.1);
+    padding-bottom: 6px;
+}
+.day-header {
+    font-size: 1.2em;
+    margin-top: 20px;
+    margin-bottom: 12px;
+    font-weight: 600;
+    color: #1976d2;
+    background-color: #f5f5f5;
+    padding: 8px 12px;
+    border-radius: 6px;
+}
+.meal-type {
+    font-weight: 600;
+    color: #d32f2f;
+    margin-right: 4px;
+}
+.markdown-text ul, .markdown-text ol {
+    margin-bottom: 12px;
+    padding-left: 24px;
+}
+.markdown-text li {
+    margin-bottom: 4px;
+}
+.markdown-text strong {
+    font-weight: 600;
+}
+.message-bubble {
+    padding: 16px;
+    border-radius: 12px;
+    max-width: 90%;
+    margin-bottom: 8px;
+}
+.user-message {
+    background-color: #e3f2fd;
+    align-self: flex-end;
+}
+.bot-message {
+    background-color: #f1f8e9;
+    align-self: flex-start;
+}
+.center-content {
+    display: flex;
+    flex-direction: column;
+    align-items: center;
+}
+.diet-card {
+    border: 1px solid #e0e0e0;
+    border-radius: 12px;
+    padding: 20px;
+    box-shadow: 0 2px 10px rgba(0, 0, 0, 0.05);
+    background-color: white;
+    margin-top: 10px;
+    margin-bottom: 20px;
+    width: 100%;
+    line-height: 1.6;
+}
+.diet-card ul li::marker {
+    color: #2e7d32;
+}
+.diet-card p {
+    margin-bottom: 8px;
+}
+.diet-list {
+    list-style-type: none;
+    padding-left: 0;
+}
+.diet-list li {
+    background-color: #f9f9f9;
+    margin-bottom: 6px;
+    padding: 6px 10px;
+    border-radius: 4px;
+    border-left: 3px solid #2e7d32;
+}
+.input-container {
+    background-color: white;
+    border-radius: 10px;
+    padding: 10px;
+    box-shadow: 0 2px 10px rgba(0, 0, 0, 0.05);
+}
+.features-container {
+    display: flex;
+    flex-wrap: wrap;
+    margin-top: 24px;
+    gap: 16px;
+}
+.feature-card {
+    flex: 1 1 calc(33% - 16px);
+    min-width: 200px;
+    background-color: white;
+    padding: 16px;
+    border-radius: 10px;
+    box-shadow: 0 2px 10px rgba(0, 0, 0, 0.05);
+}
+.feature-icon {
+    font-size: 24px;
+    margin-bottom: 10px;
+    color: #2e7d32;
+}
+.footer {
+    text-align: center;
+    margin-top: 30px;
+    font-size: 0.9em;
+    color: #666;
+}
+"""
+
+def format_markdown(text):
+    if "## Diet Overview" in text:
+        text = text.replace("## Diet Overview", '<h2 class="diet-header">Diet Overview</h2>')
+        text = text.replace("## Weekly Meal Plan", '<h2 class="diet-header">Weekly Meal Plan</h2>')
+        text = text.replace("## General Recommendations", '<h2 class="diet-header">General Recommendations</h2>')
+        
+        for day in ["Monday", "Tuesday", "Wednesday", "Thursday", "Friday", "Saturday", "Sunday"]:
+            text = text.replace(f"### {day}", f'<h3 class="day-header">{day}</h3>')
+        
+        for meal in ["Breakfast", "Lunch", "Dinner", "Snacks"]:
+            text = text.replace(f"**{meal}**:", f'<strong class="meal-type">{meal}</strong>:')
+        
+        lines = text.split('\n')
+        formatted_lines = []
+        
+        in_list = False
+        list_items = []
+        
+        for line in lines:
+            if line.strip().startswith('- '):
+                in_list = True
+                list_items.append(line.strip()[2:])
+            else:
+                if in_list:
+                    formatted_list = '<ul class="diet-list">\n'
+                    for item in list_items:
+                        formatted_list += f'  <li>{item}</li>\n'
+                    formatted_list += '</ul>'
+                    formatted_lines.append(formatted_list)
+                    in_list = False
+                    list_items = []
+                
+                formatted_lines.append(line)
+        
+        if in_list:
+            formatted_list = '<ul class="diet-list">\n'
+            for item in list_items:
+                formatted_list += f'  <li>{item}</li>\n'
+            formatted_list += '</ul>'
+            formatted_lines.append(formatted_list)
+        
+        text = '\n'.join(formatted_lines)
+        
+        text = f'<div class="diet-card markdown-text">{text}</div>'
+    
+    return text
+
+class ConversationState:
+    def __init__(self):
+        self.reset()
+    
+    def reset(self):
+        self.current_step = "greeting"
+        self.user_profile = {
+            "dietary_preference": None,
+            "health_goal": None,
+            "age": None,
+            "weight": None,
+            "height": None, 
+            "gender": None,
+            "activity_level": None,
+            "restrictions": [],
+            "allergies": []
+        }
+
+state = ConversationState()
+
+def extract_info(message):
+    """Extract all relevant information from a message at once"""
+    info = {}
+    
+    preferences = ["vegetarian", "vegan", "keto", "paleo", "gluten-free", "low-carb", 
+                  "pescatarian", "mediterranean", "dash", "plant-based", "flexitarian", 
+                  "low-fat", "carnivore", "whole food"]
+    for pref in preferences:
+        if pref in message.lower():
+            info["dietary_preference"] = pref
+            break
+    
+    goals = {
+        "lose weight": "weight loss", 
+        "weight loss": "weight loss",
+        "gain weight": "weight gain", 
+        "weight gain": "weight gain",
+        "maintain": "weight maintenance", 
+        "muscle": "muscle building",
+        "energy": "improved energy", 
+        "health": "better health"
+    }
+    for key, goal in goals.items():
+        if key in message.lower():
+            info["health_goal"] = goal
+            break
+    
+    age_match = re.search(r'(\d+)\s*(?:years|year|yr|y)(?:\s*old)?', message.lower())
+    if age_match:
+        info["age"] = age_match.group(1)
+    
+    weight_match = re.search(r'(\d+(?:\.\d+)?)\s*(?:kg|kilos?|pounds?|lbs?)', message.lower())
+    if weight_match:
+        info["weight"] = weight_match.group(0)
+    
+    height_match = re.search(r'(\d+(?:\.\d+)?)\s*(?:cm|centimeters?|meters?|m|feet|foot|ft|\'|inches|in|")', message.lower())
+    if height_match:
+        info["height"] = height_match.group(0)
+    
+    if "male" in message.lower():
+        info["gender"] = "male"
+    elif "female" in message.lower():
+        info["gender"] = "female"
+    
+    activity_keywords = {
+        "sedentary": ["sedentary", "inactive", "not active", "desk job"],
+        "lightly active": ["lightly active", "light activity", "light exercise", "walk"],
+        "moderately active": ["moderately active", "moderate activity", "moderate exercise"],
+        "very active": ["very active", "active", "highly active", "exercise regularly"]
+    }
+    
+    for level, keywords in activity_keywords.items():
+        if any(keyword in message.lower() for keyword in keywords):
+            info["activity_level"] = level
+            break
+    
+    restrictions = ["lactose intolerance", "high sodium", "added sugar", "low fodmap", 
+                   "low carb", "celiac", "gluten", "high cholesterol", "ibs"]
+    
+    found_restrictions = [r for r in restrictions if r in message.lower()]
+    if found_restrictions:
+        info["restrictions"] = found_restrictions
+    
+    allergies = ["nuts", "shellfish", "eggs", "dairy", "wheat", "soy", "fish", 
+                "peanuts"]
+    
+    found_allergies = [a for a in allergies if a in message.lower()]
+    if found_allergies:
+        info["allergies"] = found_allergies
+    
+    return info
+
+def chat_with_diet_bot(message, history):
+    if not message.strip():
+        return "", history
+    
+    if any(word in message.lower() for word in ["reset", "start over"]):
+        state.reset()
+        response = "I've reset your information. Would you like to create a diet plan?"
+        state.current_step = "request_diet_plan"
+        return "", history + [[message, response]]
+    
+    if any(word in message.lower() for word in ["goodbye", "bye", "exit"]):
+        state.reset()
+        response = "Goodbye! Feel free to come back anytime for diet advice."
+        return "", history + [[message, response]]
+
+    info = extract_info(message)
+    for key, value in info.items():
+        if key == "restrictions" or key == "allergies":
+            state.user_profile[key] = value
+        elif value:
+            state.user_profile[key] = value
+
+    if state.current_step == "greeting":
+        response = "Hello! I can help create a personalized meal plan based on your preferences and goals. Would you like me to help with that?"
+        state.current_step = "request_diet_plan"
+    
+    elif state.current_step == "request_diet_plan":
+        if any(word in message.lower() for word in ["yes", "sure", "okay", "please", "diet", "plan"]):
+            response = "Great! What type of diet are you following? (e.g., vegetarian, vegan, keto, paleo, gluten-free, or no specific diet)"
+            state.current_step = "ask_dietary_preference"
+    
+    elif state.current_step == "ask_dietary_preference":
+        if state.user_profile["dietary_preference"]:
+            response = f"Thanks for letting me know you follow a {state.user_profile['dietary_preference']} diet. What's your health goal? (e.g., weight loss, weight gain, maintenance, muscle building)"
+        else:
+            state.user_profile["dietary_preference"] = "standard balanced"
+            response = "I'll focus on a standard balanced diet plan. What's your health goal? (e.g., weight loss, weight gain, maintenance, muscle building)"
+        state.current_step = "ask_health_goal"
+    
+    elif state.current_step == "ask_health_goal":
+        if state.user_profile["health_goal"]:
+            response = "Could you share your age, weight, height, gender, and activity level? For example: '35 years old, 70kg, 175cm, male, moderate activity'"
+        else:
+            state.user_profile["health_goal"] = "general health"
+            response = "I'll focus on general health improvements. Could you share your age, weight, height, gender, and activity level?"
+        state.current_step = "ask_personal_info"
+    
+    elif state.current_step == "ask_personal_info":
+        personal_info = ["age", "weight", "height", "gender", "activity_level"]
+        missing = [item for item in personal_info if not state.user_profile[item]]
+        
+        if missing:
+            response = f"Thanks for that information. I still need your {', '.join(missing)}. Could you provide that?"
+        else:
+            response = "Do you have any dietary restrictions I should be aware of? (e.g., lactose intolerance, low sodium, low sugar)"
+            state.current_step = "ask_restrictions"
+    
+    elif state.current_step == "ask_restrictions":
+        if "no" in message.lower() or "none" in message.lower():
+            state.user_profile["restrictions"] = []
+        
+        response = "Do you have any food allergies? (e.g., nuts, shellfish, eggs, dairy)"
+        state.current_step = "ask_allergies"
+    
+    elif state.current_step == "ask_allergies":
+        if "no" in message.lower() or "none" in message.lower():
+            state.user_profile["allergies"] = []
+
+        profile = state.user_profile
+        response = (
+            f"I've collected the following information:\n"
+            f"Dietary preference: {profile['dietary_preference']}\n"
+            f"Health goal: {profile['health_goal']}\n"
+            f"Age: {profile['age']}\n"
+            f"Weight: {profile['weight']}\n"
+            f"Height: {profile['height']}\n"
+            f"Gender: {profile['gender']}\n"
+            f"Activity level: {profile['activity_level']}\n"
+            f"Restrictions: {', '.join(profile['restrictions']) if profile['restrictions'] else 'None'}\n"
+            f"Allergies: {', '.join(profile['allergies']) if profile['allergies'] else 'None'}\n\n"
+            f"Is this information correct? I'll use it to create your personalized diet plan."
+        )
+        state.current_step = "confirm_information"
+    
+    elif state.current_step == "confirm_information":
+        if any(word in message.lower() for word in ["yes", "correct", "right", "good", "ok", "okay", "fine", "sure"]):
+            response = "Thanks for confirming! I'm now generating your personalized diet plan..."
+            state.current_step = "generate_diet_plan"
+            
+            profile = state.user_profile
+            prompt = f"""
+            Create a personalized weekly meal plan based on the following user information:
+            
+            Dietary Preference: {profile['dietary_preference']}
+            Health Goal: {profile['health_goal']}
+            Age: {profile['age']}
+            Weight: {profile['weight']}
+            Height: {profile['height']}
+            Gender: {profile['gender']}
+            Activity Level: {profile['activity_level']}
+            Dietary Restrictions: {', '.join(profile['restrictions']) if profile['restrictions'] else 'None'}
+            Allergies: {', '.join(profile['allergies']) if profile['allergies'] else 'None'}
+            
+            Generate a complete meal plan with:
+            1. Diet Overview - Brief summary of the diet approach
+            2. Weekly Meal Plan - Day-by-day plan with breakfast, lunch, dinner, and snacks
+            3. General Recommendations - Additional advice and nutrition tips
+            
+            Format in Markdown with clear headers and bullet points.
+            """
+            
+            try:
+                response = co.generate(
+                    prompt=prompt,
+                    model="command",
+                    max_tokens=2000,
+                    temperature=0.7,
+                ).generations[0].text
+                
+                formatted_response = format_markdown(response)
+                response = formatted_response + "\n\nI hope this meal plan helps you achieve your goals! Feel free to ask if you have any nutrition questions."
+                state.current_step = "diet_questions"
+                
+            except Exception:
+                response = "I'm sorry, I couldn't generate a diet plan at the moment. Please try again later."
+                state.current_step = "greeting"
+        else:
+            response = "Let's update your information. What would you like to change?"
+            state.current_step = "update_information"
+    
+    elif state.current_step == "update_information":
+        profile = state.user_profile
+        response = (
+            f"I've updated your information:\n"
+            f"Dietary preference: {profile['dietary_preference']}\n"
+            f"Health goal: {profile['health_goal']}\n"
+            f"Age: {profile['age']}\n"
+            f"Weight: {profile['weight']}\n"
+            f"Height: {profile['height']}\n"
+            f"Gender: {profile['gender']}\n"
+            f"Activity level: {profile['activity_level']}\n"
+            f"Restrictions: {', '.join(profile['restrictions']) if profile['restrictions'] else 'None'}\n"
+            f"Allergies: {', '.join(profile['allergies']) if profile['allergies'] else 'None'}\n\n"
+            f"Is this information correct now?"
+        )
+        state.current_step = "confirm_information"
+    
+    elif state.current_step == "diet_questions":
+        prompt = f"""
+        Answer this nutrition question clearly and accurately:
+        Question: {message}
+        Provide helpful, evidence-based advice with practical tips.
+        """
+        
+        try:
+            response = co.generate(
+                prompt=prompt,
+                model="command",
+                max_tokens=800,
+                temperature=0.7,
+            ).generations[0].text
+        except Exception:
+            response = "I'm sorry, I couldn't answer your question at the moment. Please try again later."
+    
+    return "", history + [[message, response]]
+
+with gr.Blocks(css=custom_css, theme=gr.themes.Soft()) as demo:
+    with gr.Column():
+        gr.HTML("""
+        <div class="app-header">
+            <h1 class="app-title">Diet Planning Chatbot</h1>
+        </div>
+        """)
+        
+        gr.Markdown(
+            """
+            Hello! I can help you create a weekly meal plan based on your dietary preferences, health goals, and personal information.
+            
+            ### How to use:
+            1. Start by saying hello or asking for a diet plan
+            2. The chatbot will ask about your dietary needs, health goals, and other informations
+            3. After confirming, it will generate a personalized weekly meal plan
+            """
+        )
+        
+        chatbot = gr.Chatbot(
+            [],
+            elem_id="chatbot",
+            avatar_images=["https://api.dicebear.com/7.x/thumbs/svg?seed=user", "https://api.dicebear.com/7.x/thumbs/svg?seed=bot&backgroundColor=2e7d32"],
+            height=500,
+            render_markdown=True
+        )
+        
+        with gr.Row(elem_classes="input-container"):
+            msg = gr.Textbox(
+                show_label=False,
+                placeholder="Type your message here...",
+                scale=9,
+            )
+            clear = gr.Button("Clear", scale=1)
+        
+        msg.submit(chat_with_diet_bot, [msg, chatbot], [msg, chatbot])
+        clear.click(lambda: [], None, chatbot)
+    
+        gr.HTML("""
+        <div class="features-container">
+            <div class="feature-card">
+                <h3>Weekly Meal Plans</h3>
+                <p>Complete 7-day meal plans with breakfast, lunch, dinner, and snacks.</p>
+            </div>
+            <div class="feature-card">
+                <h3>Personalized</h3>
+                <p>Tailored to your specific dietary needs and health goals.</p>
+            </div>
+            <div class="feature-card">
+                <h3>Dietary Restrictions</h3>
+                <p>Support for vegetarian, vegan, keto, gluten-free and more.</p>
+            </div>
+        </div>
+        """)
+
+if __name__ == "__main__":
+    demo.launch(server_name="0.0.0.0") 

docker-compose.yml

@@ -0,0 +1,37 @@
+version: '3'
+
+services:
+  rasa:
+    build:
+      context: .
+      dockerfile: Dockerfile.rasa
+    container_name: rasa
+    ports:
+      - "5005:5005"
+    environment:
+      - COHERE_API_KEY=${COHERE_API_KEY}
+    volumes:
+      - ./rasa_app:/app/rasa_app
+    command: ["python", "-m", "rasa", "run", "--enable-api", "--cors", "*"]
+    healthcheck:
+      test: ["CMD", "curl", "-f", "http://localhost:5005/"]
+      interval: 10s
+      timeout: 5s
+      retries: 5
+      start_period: 15s
+
+  app:
+    build:
+      context: .
+    container_name: diet-app
+    depends_on:
+      rasa:
+        condition: service_healthy
+    ports:
+      - "7860:7860"
+    environment:
+      - COHERE_API_KEY=${COHERE_API_KEY}
+      - RASA_URL=http://rasa:5005/webhooks/rest/webhook
+    volumes:
+      - .:/app
+    command: ["python", "app.py"] 

huggingface-space.yml

@@ -0,0 +1,7 @@
+title: Diet Planning Chatbot
+emoji: 🥗
+colorFrom: green
+colorTo: blue
+sdk: docker
+app_port: 7860
+pinned: false 

rasa_app/__init__.py

@@ -0,0 +1 @@
+# Rasa app package 

rasa_app/actions/__init__.py

@@ -0,0 +1 @@
+# Actions package 

rasa_app/actions/actions.py

@@ -0,0 +1,174 @@
+from typing import Any, Text, Dict, List
+from rasa_sdk import Action, Tracker
+from rasa_sdk.executor import CollectingDispatcher
+from rasa_sdk.events import SlotSet
+import cohere
+import os
+from dotenv import load_dotenv
+import json
+import markdown
+
+load_dotenv()
+COHERE_API_KEY = os.getenv("COHERE_API_KEY")
+
+co = cohere.Client(COHERE_API_KEY)
+
+def format_user_profile(tracker):
+    dietary_preference = tracker.get_slot("dietary_preference") or "not specified"
+    health_goal = tracker.get_slot("health_goal") or "not specified"
+    age = tracker.get_slot("age") or "not specified"
+    weight = tracker.get_slot("weight") or "not specified"
+    height = tracker.get_slot("height") or "not specified"
+    gender = tracker.get_slot("gender") or "not specified"
+    activity_level = tracker.get_slot("activity_level") or "not specified"
+    restrictions = tracker.get_slot("restrictions") or []
+    allergies = tracker.get_slot("allergies") or []
+    
+    profile = {
+        "dietary_preference": dietary_preference,
+        "health_goal": health_goal,
+        "age": age,
+        "weight": weight,
+        "height": height,
+        "gender": gender,
+        "activity_level": activity_level,
+        "restrictions": restrictions,
+        "allergies": allergies
+    }
+    
+    return profile
+
+class ActionGenerateDietPlan(Action):
+    def name(self) -> Text:
+        return "action_generate_diet_plan"
+
+    def run(self, dispatcher: CollectingDispatcher,
+            tracker: Tracker,
+            domain: Dict[Text, Any]) -> List[Dict[Text, Any]]:
+        
+        user_profile = format_user_profile(tracker)
+        
+        prompt = f"""
+        Create a personalized weekly meal plan based on the following user information:
+        
+        Dietary Preference: {user_profile['dietary_preference']}
+        Health Goal: {user_profile['health_goal']}
+        Age: {user_profile['age']}
+        Weight: {user_profile['weight']}
+        Height: {user_profile['height']}
+        Gender: {user_profile['gender']}
+        Activity Level: {user_profile['activity_level']}
+        Dietary Restrictions: {', '.join(user_profile['restrictions']) if user_profile['restrictions'] else 'None'}
+        Allergies: {', '.join(user_profile['allergies']) if user_profile['allergies'] else 'None'}
+        
+        Generate a complete, well-structured meal plan with the following sections:
+        1. Diet Overview - Brief summary of the diet approach and how it aligns with their goals and preferences
+        2. Weekly Meal Plan - Detailed day-by-day plan with breakfast, lunch, dinner, and snacks
+        3. General Recommendations - Additional advice, serving sizes, and nutrition tips
+        
+        Format the response in Markdown with clear headers and bullet points.
+        """
+        
+        try:
+            response = co.generate(
+                prompt=prompt,
+                model="command",
+                max_tokens=2000,
+                temperature=0.7,
+            )
+            
+            diet_plan = response.generations[0].text
+            
+            dispatcher.utter_message(text=diet_plan)
+            
+            return [SlotSet("user_profile_complete", True)]
+        
+        except Exception as e:
+            dispatcher.utter_message(text=f"I'm sorry, I couldn't generate a diet plan at the moment. Error: {str(e)}")
+            return []
+
+class ActionAnswerDietQuestion(Action):
+    def name(self) -> Text:
+        return "action_answer_diet_question"
+
+    def run(self, dispatcher: CollectingDispatcher,
+            tracker: Tracker,
+            domain: Dict[Text, Any]) -> List[Dict[Text, Any]]:
+        
+        last_message = tracker.latest_message.get("text")
+        
+        prompt = f"""
+        Answer the following nutrition or diet-related question clearly and accurately:
+        
+        Question: {last_message}
+        
+        Provide a helpful, evidence-based response with practical advice when applicable.
+        If it would be beneficial, include a few bullet points with key takeaways.
+        """
+        
+        try:
+            response = co.generate(
+                prompt=prompt,
+                model="command",
+                max_tokens=800,
+                temperature=0.7,
+            )
+            
+            answer = response.generations[0].text
+            
+            dispatcher.utter_message(text=answer)
+            
+            return []
+        
+        except Exception as e:
+            dispatcher.utter_message(text=f"I'm sorry, I couldn't answer your question at the moment. Error: {str(e)}")
+            return []
+
+class ActionUpdateDietPlan(Action):
+    def name(self) -> Text:
+        return "action_update_diet_plan"
+
+    def run(self, dispatcher: CollectingDispatcher,
+            tracker: Tracker,
+            domain: Dict[Text, Any]) -> List[Dict[Text, Any]]:
+        
+        user_request = tracker.latest_message.get("text")
+        user_profile = format_user_profile(tracker)
+        
+        prompt = f"""
+        Update the previously generated meal plan based on the following user feedback:
+        
+        User Profile:
+        Dietary Preference: {user_profile['dietary_preference']}
+        Health Goal: {user_profile['health_goal']}
+        Age: {user_profile['age']}
+        Weight: {user_profile['weight']}
+        Height: {user_profile['height']}
+        Gender: {user_profile['gender']}
+        Activity Level: {user_profile['activity_level']}
+        Dietary Restrictions: {', '.join(user_profile['restrictions']) if user_profile['restrictions'] else 'None'}
+        Allergies: {', '.join(user_profile['allergies']) if user_profile['allergies'] else 'None'}
+        
+        User Feedback: {user_request}
+        
+        Generate an updated meal plan that addresses the user's feedback while still keeping their health goals and dietary needs in mind.
+        Format the response in Markdown with clear headers and bullet points.
+        """
+        
+        try:
+            response = co.generate(
+                prompt=prompt,
+                model="command",
+                max_tokens=2000,
+                temperature=0.7,
+            )
+            
+            updated_plan = response.generations[0].text
+            
+            dispatcher.utter_message(text=updated_plan)
+            
+            return []
+        
+        except Exception as e:
+            dispatcher.utter_message(text=f"I'm sorry, I couldn't update the diet plan at the moment. Error: {str(e)}")
+            return [] 

rasa_app/config.yml

@@ -0,0 +1,23 @@
+language: en
+
+pipeline:
+  - name: WhitespaceTokenizer
+  - name: RegexFeaturizer
+  - name: LexicalSyntacticFeaturizer
+  - name: CountVectorsFeaturizer
+  - name: CountVectorsFeaturizer
+    analyzer: "char_wb"
+    min_ngram: 1
+    max_ngram: 4
+  - name: DIETClassifier
+    epochs: 100
+  - name: EntitySynonymMapper
+  - name: ResponseSelector
+    epochs: 100
+
+policies:
+  - name: MemoizationPolicy
+  - name: TEDPolicy
+    max_history: 5
+    epochs: 100
+  - name: RulePolicy 

rasa_app/credentials.yml

@@ -0,0 +1 @@
+rest:

rasa_app/data/nlu/nlu.yml

@@ -0,0 +1,187 @@
+version: "3.1"
+
+nlu:
+- intent: greet
+  examples: |
+    - hey
+    - hello
+    - hi
+    - hello there
+    - good morning
+    - good evening
+    - hey there
+    - greetings
+    - hi there
+
+- intent: goodbye
+  examples: |
+    - bye
+    - goodbye
+    - see you around
+    - see you later
+    - bye bye
+    - later
+    - exit
+    - quit
+
+- intent: affirm
+  examples: |
+    - yes
+    - indeed
+    - of course
+    - that sounds good
+    - correct
+    - sure
+    - absolutely
+    - definitely
+    - yeah
+    - yep
+    - sounds good
+    - that's right
+    - ok
+    - okay
+    - that's correct
+    - you got it
+    - right
+    - perfect
+    - great
+
+- intent: deny
+  examples: |
+    - no
+    - never
+    - I don't think so
+    - don't like that
+    - no way
+    - not really
+    - nope
+    - negative
+    - not correct
+    - that's wrong
+    - that's not right
+    - not interested
+    - nah
+    - no thanks
+
+- intent: request_diet_plan
+  examples: |
+    - I need a diet plan
+    - Can you create a meal plan for me?
+    - I want a personalized diet
+    - I need help with meal planning
+    - Can you suggest a diet?
+    - Help me plan my meals
+    - I'm looking for nutritional advice
+    - Create a weekly meal plan for me
+    - I need a diet suggestion
+    - Can you recommend a meal plan?
+    - I want to improve my eating habits
+    - Help me eat better
+    - I need a food plan
+    - Give me a diet recommendation
+    - Plan my meals for the week
+    - What should I eat?
+    - What's a good diet for me?
+    - Can you make me a diet plan?
+
+- intent: provide_dietary_preference
+  examples: |
+    - I'm [vegetarian](dietary_preference)
+    - I follow a [vegan](dietary_preference) diet
+    - I'm on a [keto](dietary_preference) diet
+    - I eat [paleo](dietary_preference)
+    - I'm on a [gluten-free](dietary_preference) diet
+    - I don't have any specific diet
+    - My diet is [low-carb](dietary_preference)
+    - I'm [pescatarian](dietary_preference)
+    - I eat a [Mediterranean](dietary_preference) diet
+    - I follow a [DASH](dietary_preference) diet
+    - I'm on a [plant-based](dietary_preference) diet
+    - I prefer a [flexitarian](dietary_preference) approach
+    - I follow a [high-protein](dietary_preference) diet
+    - I don't follow any specific [diet](dietary_preference)
+
+- intent: provide_health_goal
+  examples: |
+    - I want to [lose weight](health_goal)
+    - I'm trying to [gain weight](health_goal)
+    - I need to [maintain my weight](health_goal)
+    - I'm looking to [build muscle](health_goal)
+    - I want to [improve my energy](health_goal)
+    - I need [better nutrition](health_goal)
+    - I want to eat for [better health](health_goal)
+    - I want to [manage my diabetes](health_goal)
+    - I'm trying to [lower my blood pressure](health_goal)
+    - I want to [reduce inflammation](health_goal)
+    - I need to [improve my gut health](health_goal)
+    - I'm looking to [enhance athletic performance](health_goal)
+    - I'm looking to [improve mental clarity](health_goal)
+    - I want to [reduce stress](health_goal)
+
+- intent: provide_personal_info
+  examples: |
+    - I'm [25](age) years old, [65](weight) kg, [170](height) cm, [male](gender), [moderate](activity_level) activity
+    - [35](age) years old, [70](weight) kg, [175](height) cm, [female](gender), [sedentary](activity_level) lifestyle
+    - I'm [40](age), weight [80](weight) kg, height [180](height) cm, [male](gender), [very active](activity_level)
+    - [19](age) years, [55](weight) kg, [165](height) cm, [female](gender), [lightly active](activity_level)
+    - I'm [30](age) years old, [75](weight) kg, [178](height) cm, [non-binary](gender), [active](activity_level)
+    - Age [45](age), weight [90](weight) kg, [185](height) cm tall, [male](gender), [sedentary](activity_level)
+    - I'm [27](age), [68](weight) kilos, [172](height) centimeters, [female](gender), [highly active](activity_level)
+    - [50](age) years, [82](weight) kg, [176](height) cm, [male](gender), [moderately active](activity_level)
+    - Age is [22](age), weight is [58](weight) kg, height is [160](height) cm, [female](gender), [active](activity_level)
+- intent: provide_restrictions
+  examples: |
+    - I have [lactose intolerance](restriction)
+    - I need to avoid [high sodium](restriction) foods
+    - I can't eat foods with [added sugar](restriction)
+    - I have to follow a [low FODMAP](restriction) diet
+    - I need [low carb](restriction) options
+    - I have [celiac disease](restriction) so I can't have gluten
+    - I need to avoid [high cholesterol](restriction) foods
+    - I have [IBS](restriction) so I need to be careful
+    - I need a [low histamine](restriction) diet
+    - I have to watch my [oxalate](restriction) intake
+    - I'm on a [low residue](restriction) diet
+    - I need a [low acidic](restriction) diet
+    - I have [fructose malabsorption](restriction)
+    - I need [kidney-friendly](restriction) options
+    - I require a [renal diet](restriction)
+    - I have [diverticulitis](restriction)
+    - I'm on a [low iodine](restriction) diet
+
+- intent: provide_allergies
+  examples: |
+    - I'm allergic to [nuts](allergy)
+    - I have a [shellfish](allergy) allergy
+    - I can't eat [eggs](allergy)
+    - I have a [dairy](allergy) allergy
+    - I'm allergic to [wheat](allergy)
+    - I can't have [soy](allergy)
+    - I have a [fish](allergy) allergy
+    - I'm allergic to [peanuts](allergy)
+    - I can't eat [sesame](allergy)
+    - I have an [almond](allergy) allergy
+    - I'm allergic to [mustard](allergy)
+    - I'm allergic to [molluscs](allergy)
+    - I can't eat [sulfites](allergy)
+    - I have a [corn](allergy) allergy
+    - I have a [banana](allergy) allergy
+    - I'm allergic to [latex fruits](allergy)
+
+- intent: ask_diet_question
+  examples: |
+    - What foods are high in protein?
+    - How many calories should I consume daily?
+    - Is intermittent fasting effective?
+    - What's the best time to eat before a workout?
+    - Are eggs healthy?
+    - How much water should I drink daily?
+    - What are good sources of fiber?
+    - What should I eat before bed?
+    - How can I increase my metabolism?
+    - What's the healthiest cooking oil?
+    - How much protein do I need daily?
+    - What supplements should I take?
+    - What fruits are lowest in sugar?
+    - Is dairy necessary for calcium?
+    - What's the best diet for heart health?

rasa_app/data/rules/rules.yml

@@ -0,0 +1,28 @@
+version: "3.1"
+
+rules:
+- rule: Say goodbye anytime the user says goodbye
+  steps:
+  - intent: goodbye
+  - action: utter_goodbye
+
+- rule: Say hello anytime the user says hello
+  steps:
+  - intent: greet
+  - action: utter_greet
+
+- rule: Answer diet questions
+  steps:
+  - intent: ask_diet_question
+  - action: action_answer_diet_question
+
+- rule: Provide recommendation when asked
+  steps:
+  - intent: ask_for_recommendation
+  - action: action_answer_diet_question
+
+- rule: Handle changes to diet plan
+  steps:
+  - intent: ask_for_changes
+  - action: utter_acknowledge_changes
+  - action: action_update_diet_plan 

rasa_app/data/stories/stories.yml

@@ -0,0 +1,91 @@
+version: "3.1"
+
+stories:
+- story: happy path - dietary planning
+  steps:
+  - intent: greet
+  - action: utter_greet
+  - intent: request_diet_plan
+  - action: utter_ask_dietary_preference
+  - intent: provide_dietary_preference
+  - action: utter_ask_health_goal
+  - intent: provide_health_goal
+  - action: utter_ask_personal_info
+  - intent: provide_personal_info
+  - action: utter_ask_restrictions
+  - intent: provide_restrictions
+  - action: utter_ask_allergies
+  - intent: provide_allergies
+  - action: utter_confirm_information
+  - intent: affirm
+  - action: utter_generating_plan
+  - action: action_generate_diet_plan
+  - intent: ask_for_changes
+  - action: utter_ask_for_changes
+  - intent: provide_dietary_preference
+  - action: utter_acknowledge_changes
+  - action: action_update_diet_plan
+
+- story: direct diet plan request
+  steps:
+  - intent: request_diet_plan
+  - action: utter_ask_dietary_preference
+  - intent: provide_dietary_preference
+  - action: utter_ask_health_goal
+  - intent: provide_health_goal
+  - action: utter_ask_personal_info
+  - intent: provide_personal_info
+  - action: utter_ask_restrictions
+  - intent: provide_restrictions
+  - action: utter_ask_allergies
+  - intent: provide_allergies
+  - action: utter_confirm_information
+  - intent: affirm
+  - action: utter_generating_plan
+  - action: action_generate_diet_plan
+
+- story: diet plan with no restrictions or allergies
+  steps:
+  - intent: greet
+  - action: utter_greet
+  - intent: request_diet_plan
+  - action: utter_ask_dietary_preference
+  - intent: provide_dietary_preference
+  - action: utter_ask_health_goal
+  - intent: provide_health_goal
+  - action: utter_ask_personal_info
+  - intent: provide_personal_info
+  - action: utter_ask_restrictions
+  - intent: deny
+  - action: utter_ask_allergies
+  - intent: deny
+  - action: utter_confirm_information
+  - intent: affirm
+  - action: utter_generating_plan
+  - action: action_generate_diet_plan
+
+- story: ask diet question
+  steps:
+  - intent: ask_diet_question
+  - action: action_answer_diet_question
+
+- story: ask for recommendation
+  steps:
+  - intent: ask_for_recommendation
+  - action: action_answer_diet_question
+
+- story: reject generated plan and request changes
+  steps:
+  - action: action_generate_diet_plan
+  - intent: deny
+  - action: utter_ask_for_changes
+  - intent: ask_for_changes
+  - action: utter_acknowledge_changes
+  - action: action_update_diet_plan
+
+- story: greet and goodbye
+  steps:
+  - intent: greet
+  - action: utter_greet
+  - intent: goodbye
+  - action: utter_goodbye 

rasa_app/domain.yml

@@ -0,0 +1,138 @@
+version: "3.1"
+
+intents:
+  - greet
+  - goodbye
+  - affirm
+  - deny
+  - request_diet_plan
+  - provide_dietary_preference
+  - provide_health_goal
+  - provide_personal_info
+  - provide_restrictions
+  - provide_allergies
+  - ask_diet_question
+  - ask_for_changes
+  - ask_for_recommendation
+
+entities:
+  - dietary_preference
+  - health_goal
+  - age
+  - weight
+  - height
+  - gender
+  - activity_level
+  - restriction
+  - allergy
+
+slots:
+  dietary_preference:
+    type: text
+    influence_conversation: true
+    mappings:
+      - type: from_entity
+        entity: dietary_preference
+  health_goal:
+    type: text
+    influence_conversation: true
+    mappings:
+      - type: from_entity
+        entity: health_goal
+  age:
+    type: text
+    influence_conversation: false
+    mappings:
+      - type: from_entity
+        entity: age
+  weight:
+    type: text
+    influence_conversation: false
+    mappings:
+      - type: from_entity
+        entity: weight
+  height:
+    type: text
+    influence_conversation: false
+    mappings:
+      - type: from_entity
+        entity: height
+  gender:
+    type: text
+    influence_conversation: false
+    mappings:
+      - type: from_entity
+        entity: gender
+  activity_level:
+    type: text
+    influence_conversation: false
+    mappings:
+      - type: from_entity
+        entity: activity_level
+  restrictions:
+    type: list
+    influence_conversation: true
+    mappings:
+      - type: from_entity
+        entity: restriction
+  allergies:
+    type: list
+    influence_conversation: true
+    mappings:
+      - type: from_entity
+        entity: allergy
+  user_profile_complete:
+    type: bool
+    initial_value: false
+    influence_conversation: true
+    mappings:
+      - type: custom
+
+responses:
+  utter_greet:
+    - text: "Hello! I'm your diet planning assistant. I can help you create a personalized meal plan. Would you like me to create a diet plan for you?"
+
+  utter_goodbye:
+    - text: "Goodbye! Feel free to come back anytime for diet advice."
+
+  utter_ask_dietary_preference:
+    - text: "What type of diet are you following? (e.g., vegetarian, vegan, keto, paleo, gluten-free, or no specific diet)"
+
+  utter_ask_health_goal:
+    - text: "What's your health goal? (e.g., weight loss, weight gain, maintenance, muscle building, improved energy)"
+
+  utter_ask_personal_info:
+    - text: "To provide the most accurate recommendations, I need some basic information. Could you share your age, weight, height, gender, and activity level? For example: '35 years old, 70kg, 175cm, male, moderate activity'"
+
+  utter_ask_restrictions:
+    - text: "Do you have any dietary restrictions I should be aware of? (e.g., lactose intolerance, low sodium, low sugar)"
+
+  utter_ask_allergies:
+    - text: "Do you have any food allergies? (e.g., nuts, shellfish, eggs, dairy)"
+
+  utter_confirm_information:
+    - text: "I've collected the following information:\nDietary preference: {dietary_preference}\nHealth goal: {health_goal}\nAge: {age}\nWeight: {weight}\nHeight: {height}\nGender: {gender}\nActivity level: {activity_level}\nRestrictions: {restrictions}\nAllergies: {allergies}\n\nIs this information correct? I'll use it to create your personalized diet plan."
+
+  utter_generating_plan:
+    - text: "Thanks for confirming! I'm now generating your personalized diet plan based on your information. This will just take a moment..."
+
+  utter_ask_diet_question:
+    - text: "I'd be happy to answer your diet-related questions. What would you like to know?"
+
+  utter_ask_for_changes:
+    - text: "Would you like to make any changes to the diet plan I've suggested?"
+
+  utter_acknowledge_changes:
+    - text: "I'll adjust the diet plan based on your feedback. Let me update it for you."
+
+  utter_default:
+    - text: "I'm not sure I understand. Could you rephrase that?"
+
+actions:
+  - action_generate_diet_plan
+  - action_answer_diet_question
+  - action_update_diet_plan
+
+session_config:
+  session_expiration_time: 60
+  carry_over_slots_to_new_session: true 

rasa_app/endpoints.yml

@@ -0,0 +1,2 @@
+action_endpoint:
+  url: "http://localhost:5055/webhook" 

requirements.txt

@@ -0,0 +1,5 @@
+gradio==3.41.0
+cohere==4.27
+python-dotenv==1.0.0
+requests==2.31.0
+markdown==3.4.3 

run.sh

@@ -0,0 +1,15 @@
+#!/bin/bash
+
+if [ ! -f .env ]; then
+    cp .env.example .env
+    exit 1
+fi
+
+cd rasa_app && rasa run --enable-api --cors "*" &
+RASA_PID=$!
+
+sleep 5
+
+cd .. && python app.py
+
+trap "kill $RASA_PID" EXIT 

train_rasa.py

@@ -0,0 +1,31 @@
+import os
+import subprocess
+import sys
+
+def train_rasa_model():
+    
+    os.chdir('./rasa_app')
+    
+    try:
+        result = subprocess.run(
+            ["python", "-m", "rasa", "train"],
+            check=True,
+            stdout=subprocess.PIPE,
+            stderr=subprocess.PIPE,
+            text=True
+        )
+        
+        if result.returncode == 0:
+            return True
+        else:
+            return False
+            
+    except subprocess.CalledProcessError as e:
+        return False
+    except Exception as e:
+        return False
+
+if __name__ == "__main__":
+    success = train_rasa_model()
+    if not success:
+        sys.exit(1)