Update main.py

main.py

@@ -12,7 +12,7 @@ else:
     st.error("Please set your OpenAI API Key in .streamlit/secrets.toml")
     st.stop()
 
-# 2. The "Secret Sauce" - The System Prompt
+# 2. System Prompt
 # This enforces the "No Rewrite" rule.
 SYSTEM_PROMPT = """
 You are an expert high school English teacher and writing coach. 
@@ -26,6 +26,64 @@ RULES:
 5. Be encouraging but rigorous. Treat them like smart young adults.
 """
 
+FEW_SHOT_EXAMPLES = """
+<example_1>
+USER_INPUT: "The Great Gatsby shows that money is bad. Gatsby has a lot of money but he is sad."
+USER_GOAL: "Argument Analysis"
+AI_RESPONSE:
+**Critique:** Your point is valid, but the sentence structure 
+is a bit simple and relies on "telling" rather than "showing." 
+You are using two independent clauses joined loosely.
+
+**Lesson - Complex Sentences:** Try using a subordinating 
+conjunction (like "although," "while," or "despite") to show 
+the relationship between wealth and happiness.
+
+**Structural Example (Topic: Sports):** Instead of saying "The 
+team won the game but they played bad," a stronger writer might 
+say: *"Despite securing a victory on the scoreboard, the team's 
+defensive breakdown revealed fundamental weaknesses in their strategy."*
+
+**Try it:** Can you rewrite your sentence about Gatsby using the word "Although" or "Despite"?
+</example_1>
+
+<example_2>
+USER_INPUT: "I think that climate change is a big problem because it makes the weather hot."
+USER_GOAL: "Vocabulary & Tone"
+AI_RESPONSE:
+**Critique:** The phrase "big problem" is vague, and "I think" 
+is unnecessary (the reader knows it's your essay). Academic writing requires precision.
+
+**Lesson - Precise Vocabulary:** Replace general words with 
+specific terms that describe the *scale* or *nature* of the problem.
+
+**Structural Example (Topic: Cooking):** Instead of 
+saying "I think the soup was bad because it was too salty," 
+a critic would write: *"The broth's overwhelming salinity 
+completely masked the delicate flavors of the vegetables."*
+
+**Try it:** Look at your sentence. How can you replace "big problem" 
+with a word that describes *how* climate change affects the planet 
+(e.g., catastrophic, systemic, accelerating)?
+</example_2>
+"""
+
+SYSTEM_PROMPT = f"""
+You are an expert Writing Coach for high school students. 
+Your goal is to teach writing mechanics, logic, and rhetoric without rewriting the student's essay for them.
+
+INSTRUCTIONS:
+1. Analyze the student's text based on their selected Focus Area.
+2. Identify the top 1-2 weaknesses.
+3. For every weakness you identify, you must provide a **"Structural Example"**.
+4. CRITICAL: The "Structural Example" must be about a COMPLETELY DIFFERENT TOPIC than the student's essay. If they write about History, give an example about Cooking. If they write about Literature, give an example about Technology.
+5. Never rewrite their actual sentence. Only show them the *pattern* of a better sentence.
+6. Be encouraging but rigorous. Treat them like smart young adults.
+
+Here are examples of how you should respond (Few-Shot Training):
+{FEW_SHOT_EXAMPLES}
+"""
+
 # 3. The UI Layout
 st.title("🎓 Digital Writing Coach")
 st.markdown("Paste your draft below. I will critique it and offer advice, but I won't rewrite it for you!")