MAIN_PROMPT = """
### **Module 4: Proportional Thinking with Percentages**  
"Welcome to this module on proportional reasoning with percentages!
Your goal is to solve a real-world problem using different representations:
1️⃣ **Bar Model**
2️⃣ **Double Number Line**
3️⃣ **Equation-Based Approach**

🚀 **Here’s the problem:**  
Orrin and Damen decided to invest money in a local ice cream shop. Orrin invests $1,500, which is 60% of their total investment. How much do Orrin and Damen invest together?

💡 **Before receiving guidance, choose a method and explain your reasoning.**  
🚀 **Which method would you like to use first?**  
(Type 'Bar Model,' 'Double Number Line,' or 'Equation' to proceed.)"
"""

def get_prompt_for_method(method):
    if method.lower() == "bar model":
        return """
### **Bar Model Approach**  
Great choice! The Bar Model is a useful way to visualize proportions and percentages.  

📌 **Now, please apply the Bar Model and explain your approach.**  
- How would you represent the total investment in the bar model?  
- How would you use it to find the unknown amount?  

✏️ **Go ahead and describe your approach first.** I will provide feedback after hearing your reasoning.
"""

    elif method.lower() == "double number line":
        return """
### **Double Number Line Approach**  
Great choice! The Double Number Line helps align percentage values with real-world quantities.  

📌 **Now, please apply the Double Number Line and explain your approach.**  
- How would you structure the number lines?  
- How would you align percentages with dollar values?  

✏️ **Go ahead and describe your approach first.** I will provide feedback after hearing your reasoning.
"""

    elif method.lower() == "equation":
        return """
### **Equation-Based Approach**  
Great choice! Setting up an equation is a powerful way to represent proportional relationships.  

📌 **Now, please apply the Equation method and explain your approach.**  
- How would you write an equation to represent this problem?  
- What steps would you take to solve for the unknown?  

✏️ **Go ahead and describe your approach first.** I will provide feedback after hearing your reasoning.
"""

    return "I didn’t understand your choice. Please type 'Bar Model,' 'Double Number Line,' or 'Equation' to proceed."


def get_feedback_for_method(method, teacher_response):
    teacher_response = teacher_response.lower().strip()  # Normalize input for better matching

    if method.lower() == "bar model":
        if "divide" in teacher_response and "60%" in teacher_response:
            return "Great start! You recognized that the bar should be divided into parts representing percentages. Now, can you calculate how much each part represents?"
        elif "10%" in teacher_response or "one part is 10%" in teacher_response:
            return "Nice work! Each part represents 10% of the total. Now, how much does one part represent in dollars?"
        elif "250" in teacher_response:
            return "Correct! Each part is worth $250. Now, how can you use this to determine the total investment?"
        else:
            return "You're close! Remember, the bar represents the total investment. Try dividing it into 10 equal parts, with 6 parts representing Orrin’s 60%. What would one part represent in percentage and dollars?"
    
    elif method.lower() == "double number line":
        if "label" in teacher_response and "percentages" in teacher_response:
            return "Nice work! You’ve set up the number line correctly. Can you now align the percentage values with the corresponding dollar amounts?"
        elif "10%" in teacher_response or "one part is 10%" in teacher_response:
            return "Good thinking! Each section represents 10% of the total investment. Now, how much is 10% in dollars?"
        elif "250" in teacher_response:
            return "That's right! Each section is worth $250. Now, can you find the total investment?"
        else:
            return "Try labeling your number line with 0%, 60%, and 100% on one side and the corresponding dollar amounts on the other. How do the values align?"

    elif method.lower() == "equation":
        if "60/100" in teacher_response and "$1500/x" in teacher_response:
            return "You're on the right track! Now, how can you solve for x in your equation?"
        elif "cross multiply" in teacher_response:
            return "Yes! Using cross multiplication will help. What do you get when solving for x?"
        elif "2500" in teacher_response:
            return "Great! The total investment is $2,500. Would you like to reflect on how the equation helped in solving this?"
        else:
            return "Try writing the proportion as (60/100) = (1500/x). What steps would you take to solve for x?"
    
    return "Interesting approach! Could you clarify your reasoning a bit more?"