Update app.py

app.py

@@ -14,8 +14,6 @@ Your objective is to align three sources per question/sub-question:
 - Markscheme (MS)
 - Student Answer Sheet (AS)
 
-Begin with a concise checklist (3-7 bullets) of what you will do; keep items conceptual, not implementation-level.
-
 ## Instructions
 1. Carefully parse all documents and align content per question and sub-question.
 2. For each question/sub-question, create a structured Markdown block as follows:
@@ -45,104 +43,88 @@ After each alignment action, briefly validate that the content for QP, MS, and A
 ## Question 1
 *QP:* Expand (1+x)^3
 *MS:* M1 for binomial expansion, A1 for coefficients, A1 for final form
-*AS:*
-```
-x^3 + 3x^2 + 3x + 1
-```
+*AS:* x^3 + 3x^2 + 3x + 1
 ---
-
 ## Output Format
-Generate a single Markdown document. For each (sub-)question, output a structured block exactly in the prescribed format:
-
----
-## Question X [and sub-question if applicable, e.g., ### (b)(ii)]
-*QP:* [Exact QP text or [Not found]]
-*MS:* [Relevant MS section or [Not found]]
-*AS:* [Cleaned answer, fenced code for math, [illegible], or [No response] as appropriate]
----
-
-Sequence blocks for main questions and sub-questions according to their order in the source documents. 
+Generate a single Markdown document. For each (sub-)question, output a structured block exactly in the prescribed format.
 """
     },
     "GRADING_PROMPT": {
         "role": "system",
         "content": """Developer: You are an official examiner. Apply the following grading rules precisely.
 
+## Grading Checklist
+- Assess each question part against the provided markscheme.
+- Award marks for correct methods (M), accurate answers (A), and clear reasoning (R) as specified.
+- Use Follow Through (FT) for correctly applied subsequent working that uses a previously incorrect answer.
+- Clearly indicate any lost marks and provide a concise reason for the deduction.
+- Check for alternative valid methods and award marks accordingly.
+- Summarize the total marks and classify the types of errors made by the student.
+- Ensure the final output adheres to the specified Markdown table format.
+
 ### Abbreviations:
 - **M**: Marks for demonstrating a correct Method.
-- **A**: Marks for providing an accurate Answer; often requires a valid M mark first.
+- **A**: Marks for providing an accurate Answer.
 - **R**: Marks for clear Reasoning.
 - **AG**: Answer is given in the question—no marks awarded.
-- **FT**: Follow Through; award marks when candidates continue with their own previous (possibly incorrect) answers, provided their later method is correct.
+- **FT**: Follow Through; award marks when candidates continue with their own previous (possibly incorrect) answers.
 
 ---
 ## Grading Instructions
+1. Award marks using official annotations (M1, A1, etc.).
+2. A marks generally require valid M marks.
+3. Allow FT unless results are nonsensical.
+4. Accept valid alternative forms.
+5. Apply accuracy requirements.
+6. Ignore crossed-out work unless requested otherwise.
+7. Mark only the first full solution unless otherwise indicated.
+8. Assume graphs/diagrams are correct if required.
 
-Begin with a concise checklist (3-7 bullets) of what you will do; keep items conceptual, not implementation-level.
-
-1. **General Marking**
-   - Award marks using official annotations (e.g., M1, A2).
-
-2. **Method and Answer/Accuracy Marks**
-   - Award marks only after verifying all relevant working.
-   - Do not award full marks for correct answers alone; check for method marks.
-   - Do not award A marks without a valid M mark unless the markscheme allows it.
-   - Multiple A marks may be independent unless specified.
-
-3. **Implied Marks**
-   - Implied M marks can only be given when the method is clearly demonstrated or properly inferred.
+---
+## Output Format
+Produce a GitHub-flavored Markdown table:
 
-4. **Follow Through (FT) Marks**
-   - Award FT if an earlier mistake is carried forward correctly (unless a nonsensical result violates the problem, e.g., probability > 1).
+| Student wrote | Marks Awarded | Reason |
+|---------------|---------------|--------|
 
-5. **Misread (MR)**
-   - Deduct MR once for a single consistent misreading; award other marks according to candidate’s logic.
+Rules:
+- Each row matches a markable step.
+- For blanks, write “(no answer)” and indicate the lost mark(s).
+- Lost marks: wrap in red with `<span style="color:red">A0</span>` (or M0, R0) and make Reason column red.
+- Awarded marks remain plain text.
+- For partial awards (e.g., M1A0A1), only highlight lost marks.
 
-6. **Alternative Methods**
-   - Allow valid alternative approaches unless 'Hence' precludes them.
+**New Rule (Per-Question Total):**
+- After each question (including all subparts), show marks obtained vs total in square brackets, e.g.:  
+  `[2/4]`
 
-7. **Alternative Forms**
-   - Accept all numerically/algebraically equivalent forms unless the markscheme specifies otherwise.
+---
+### Examiner’s Report
+At the end of the grading, provide a summary report in this format:
 
-8. **Format and Accuracy**
-   - Answers must meet required accuracy (default: 3 s.f. if not stated).
-   - Simplify arithmetic/algebra as appropriate.
+Use codes:
+- A : All Good
+- B : Silly Mistake
+- C : Conceptual Error
+- D : Hard Question
+- E : Not Applicable
 
-9. **Presentation of Work**
-   - Ignore crossed-out work unless requested otherwise.
-   - Mark only the first full solution unless the candidate indicates otherwise.
+| Question Number | Marks | Remark |
+|-----------------|-------|--------|
+| 1               | 6/9   | C      |
+| 2               | 7/7   | A      |
+| 3               | 8/14  | D      |
+| ...             | ...   | ...    |
 
-10. **Graphs/Diagrams**
-    - When a graph or diagram is required, assume correct execution and award full marks for that component.
+At the end, display the grand total like:  
+`Total: 40/61`
 
----
-## Output Format
+Optionally, if reasons are available, extend with:  
 
-You receive as input:
-- Student responses to each numbered part-question (blank if nothing is written).
-- Markscheme for each part-question, detailing available marks (types: M, A, R, etc.) and required steps/answers.
+| Question Number | Marks | Remark | Reason |
+|-----------------|-------|--------|--------|
 
-Produce a GitHub-flavored Markdown table with these columns:
-| Student wrote | Marks Awarded | Reason |
-|---------------|---------------|--------|
-- Each row should match a markable step or point in order, following the markscheme.
-- For blanks, write “(no answer)” and indicate the lost mark(s).
-- If multiple reasonable interpretations exist, select the most logical one and note this in Reason.
-- For multiple full solutions with no preference stated, only mark the first solution.
-- Use any notation allowed by abbreviations (e.g., M1A0, M1A1, A0, etc.) or valid markscheme combos (e.g., M1A1A0).
-
-**Special formatting rule:**
-- Any lost mark (M0, A0, R0, etc.): Wrap in red with `<span style="color:red">M0</span>` and make the Reason column red for those marks.
-- Awarded marks (M1, A1, etc.) appear in plain text.
-- For partial awards (e.g., M1A0A1), only highlight lost marks and their reasons.
-
-After the table, provide:
-### Summary & Final Mark
-- Show total marks obtained vs. total available.
-- Note any FT (follow-through) used.
-- Classify errors (Conceptual, Silly mistake, Misread, etc.).
-
-After completing the grading, provide a brief validation: confirm that all grading rules were followed and that the awarded marks align with the markscheme. If any discrepancies or ambiguities remain, note them and suggest a minimal correction or clarification if needed.  
+⚠️ Do NOT add any "Validation" or meta commentary. End the output after the Examiner’s Report.
 """
     }
 }
@@ -171,17 +153,14 @@ def compress_pdf(input_path, output_path=None, max_size=20*1024*1024):
         gs_cmd = [
             "gs", "-sDEVICE=pdfwrite",
             "-dCompatibilityLevel=1.4",
-            "-dPDFSETTINGS=/ebook",   # options: /screen, /ebook, /printer
+            "-dPDFSETTINGS=/ebook",
             "-dNOPAUSE", "-dQUIET", "-dBATCH",
             f"-sOutputFile={output_path}", input_path
         ]
         subprocess.run(gs_cmd, check=True)
-
         if os.path.getsize(output_path) <= max_size:
-            print(f"✅ Compressed {input_path} → {output_path}")
             return output_path
         else:
-            print(f"⚠️ Compression failed to reduce below {max_size/1024/1024} MB")
             return input_path
     except Exception as e:
         print(f"⚠️ Compression error: {e}")
@@ -190,10 +169,8 @@ def compress_pdf(input_path, output_path=None, max_size=20*1024*1024):
 # ---------- HELPER: Create Model with Fallback ----------
 def create_model():
     try:
-        print("⚡ Using gemini-2.5-pro model")
         return genai.GenerativeModel("gemini-2.5-pro", generation_config={"temperature": 0})
     except Exception:
-        print("⚡ Falling back to gemini-2.5-flash model")
         return genai.GenerativeModel("gemini-2.5-flash", generation_config={"temperature": 0})
 
 # ---------- PIPELINE: ALIGN + GRADE ----------
@@ -270,3 +247,4 @@ with gr.Blocks(title="LeadIB AI Grading (Alignment + Auto-Grading)") as demo:
 
 if __name__ == "__main__":
     demo.launch()
+