Update app.py

app.py

@@ -21,35 +21,26 @@ client = OpenAI(api_key=api_key)
 
 # 2. System Prompt
 # This enforces the "No Rewrite" rule.
-SYSTEM_PROMPT = """
-You are an expert high school English teacher and writing coach. 
-Your goal is to help students improve their writing skills, NOT to do the work for them.
-
-RULES:
-1. DO NOT rewrite the student's text. 
-2. If you see a grammatical error, quote the sentence and explain the grammar rule they broke.
-3. If you see weak argumentation, ask a Socratic question to prompt deeper thinking.
-4. Structure your feedback in Markdown with clear headings: "General Feedback", "Strengths", and "Areas for Improvement".
-5. Be encouraging but rigorous. Treat them like smart young adults.
-"""
-
+# --- 1. DEFINE THE FEW-SHOT EXAMPLES FIRST ---
 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.
+**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.
+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."*
+**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>
@@ -58,34 +49,50 @@ defensive breakdown revealed fundamental weaknesses in their strategy."*
 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)?
+**Critique:** The phrase "big problem" is vague, 
+and "I think" is unnecessary. 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?
 </example_2>
 """
 
+# --- 2. DEFINE THE MASTER SYSTEM PROMPT ---
+# We inject the few-shot examples at the end.
 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:
+CORE RULES:
+1. DO NOT rewrite the student's text. 
+2. If you see a grammatical error, quote the sentence and explain the grammar rule they broke.
+3. Structure your feedback in Markdown with clear headings: "General Feedback", "Strengths", and "Areas for Improvement".
+4. Be encouraging but rigorous. Treat them like smart young adults.
+
+SOCRATIC INSTRUCTIONS (Use when critiquing logic/argument):
+1. Do not give the answer.
+2. Ask a question that exposes the gap in the student's reasoning.
+3. Use the "Counter-Factual" technique: "If X were true, wouldn't Y also happen?"
+4. Use the "Perspective Shift" technique: "How would a French soldier in 1812 respond to this claim?"
+
+INSTRUCTIONS FOR EXAMPLES:
 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.
+4. CRITICAL: The "Structural Example" must be about a COMPLETELY DIFFERENT TOPIC than the student's essay.
 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}
@@ -128,11 +135,32 @@ if analyze_btn and user_text:
             )
             
             feedback = response.choices[0].message.content
+
+            # --- COST TRACKING LOGIC ---
+            # Extract usage data directly from the API response
+            usage = response.usage
+            input_tokens = usage.prompt_tokens
+            output_tokens = usage.completion_tokens
+            total_tokens = usage.total_tokens
+            
+            # GPT-4o Pricing (Estimate as of Late 2025)
+            # Input: $2.50 / 1M tokens
+            # Output: $10.00 / 1M tokens
+            input_cost = (input_tokens / 1_000_000) * 2.50
+            output_cost = (output_tokens / 1_000_000) * 10.00
+            total_cost = input_cost + output_cost
             
             # 5. Display Feedback
             st.markdown("---")
             st.subheader("📝 Coach's Feedback")
             st.markdown(feedback)
+
+            # --- DISPLAY METRICS (In a sidebar or expandable box) ---
+            with st.expander("📊 Session Metrics & Cost"):
+                col_a, col_b, col_c = st.columns(3)
+                col_a.metric("Input Tokens", input_tokens)
+                col_b.metric("Output Tokens", output_tokens)
+                col_c.metric("Est. Cost", f"${total_cost:.5f}") # 5 decimal places for micro-costs
             
         except Exception as e:
             st.error(f"An error occurred: {e}")