Update app.py

app.py

@@ -9,6 +9,7 @@ from typing import List, Dict, Optional
 import os
 from pathlib import Path
 import time
+import fitz
 
 GRADING_RUBRIC = """
 DETAILED GRADING RUBRIC & INSTRUCTIONS
@@ -281,6 +282,13 @@ Your output should be a single JSON format structure response as indicated by th
 **Use the following rubric to grade the essay:**
 {GRADING_RUBRIC}
 
+**For your reference, here is an example of a perfectly formatted paper. Use it as a guide for what correct formatting looks like:**
+---
+**EXAMPLE PAPER START**
+{EXAMPLE_PAPER}
+**EXAMPLE PAPER END**
+---
+
 **Instructions:**
 1.  Read the entire essay provided below.
 2.  Assess the essay against each category in the rubric.
@@ -346,6 +354,20 @@ class EssayParser:
             raise IOError(
                 f"Could not read file: {os.path.basename(file_path)}. Error: {e}")
 
+    @staticmethod
+    def parse_pdf(file_path: str) -> str:
+        """Extracts all text from a PDF document."""
+        try:
+            doc = fitz.open(file_path)
+            text = ""
+            for page in doc:
+                text += page.get_text()
+            return text
+        except Exception as e:
+            raise IOError(
+                f"Could not read PDF file: {os.path.basename(file_path)}. Error: {e}")
+
+
 
 class GeminiGrader:
     """Manages interaction with the Google Gemini API for grading."""
@@ -382,6 +404,12 @@ class GeminiGrader:
         Sends the essay to Gemini for grading and parses the JSON response.
         This is a synchronous method designed to be run in a thread pool.
         """
+        # Inject the rubric and the example paper text into the main prompt template
+        final_prompt = BASE_PROMPT_TEMPLATE.format(
+            GRADING_RUBRIC=GRADING_RUBRIC,
+            EXAMPLE_PAPER=example_paper_text
+        )
+        
         prompt_with_essay = f"{GEMINI_PROMPT}\n{essay_text}"
         try:
             response = self.model.generate_content(prompt_with_essay)
@@ -421,7 +449,7 @@ class GeminiGrader:
 # --- Gradio Application ---
 
 async def grade_papers_concurrently(
-    files: List[gr.File], api_key: str, progress=gr.Progress(track_tqdm=True)
+    files: List[gr.File], example_paper_file: gr.File, api_key: str, progress=gr.Progress(track_tqdm=True)
 ) -> (str, str):
     """
     The main asynchronous function that orchestrates the grading process.
@@ -434,6 +462,15 @@ async def grade_papers_concurrently(
     if not files:
         raise gr.Error("Please upload at least one Word document.")
 
+    # Process the optional example paper
+    example_paper_text = "No example paper provided."
+    if example_paper_file:
+        try:
+            progress(0, desc="Parsing example paper...")
+            example_paper_text = EssayParser.parse_pdf(example_paper_file.name)
+        except IOError as e:
+            raise gr.Error(f"Failed to process example PDF: {e}")
+
     try:
         grader = GeminiGrader(api_key)
     except ValueError as e:
@@ -451,7 +488,8 @@ async def grade_papers_concurrently(
                 executor,
                 process_single_file,
                 file_path,
-                grader
+                grader,
+                example_paper_text
             )
             for file_path in file_paths
         ]
@@ -502,7 +540,7 @@ async def grade_papers_concurrently(
     return output_markdown, f"{status}\n{runtime}"
 
 
-def process_single_file(file_path: str, grader: GeminiGrader) -> GradingResult:
+def process_single_file(file_path: str, grader: GeminiGrader, example_paper_text: str) -> GradingResult:
     """
     Synchronous wrapper function to parse and grade one file.
     This function is what runs in each thread of the ThreadPoolExecutor.
@@ -516,7 +554,7 @@ def process_single_file(file_path: str, grader: GeminiGrader) -> GradingResult:
                 success=False,
                 error_message="The document is empty or contains no readable text."
             )
-        return grader.grade_essay(essay_text, file_name)
+        return grader.grade_essay(essay_text, file_name, example_paper_text)
     except Exception as e:
         return GradingResult(file_name=file_name, success=False, error_message=str(e))
 
@@ -529,11 +567,11 @@ with gr.Blocks(theme=gr.themes.Soft(), title="Nursing Essay Grader") as demo:
         # 📝 Gemini-Powered Nursing Essay Grader
         Upload one or more student essays in Word format (`.docx`) to have them graded by AI.
         1.  Enter your Google API Key (enabling the Gemini API in your Google Cloud project is required).
-        2.  Upload the `.docx` files.
-        3.  Click "Grade All Papers". The results will appear below.
+        2.  Upload the `.docx` files you want to grade.
+        3.  Optionally, upload a single `.pdf` file as a "perfect" example for the AI to reference.
+        4.  Click "Grade All Papers". The results will appear below.
         """
     )
-
     with gr.Row():
         api_key_input = gr.Textbox(
             label="Google API Key",
@@ -541,21 +579,26 @@ with gr.Blocks(theme=gr.themes.Soft(), title="Nursing Essay Grader") as demo:
             type="password",
             scale=1
         )
-
-    file_uploads = gr.File(
-        label="Upload Word Document Essays",
-        file_count="multiple",
-        file_types=[".docx"],
-        type="filepath"  # Use filepath for easier handling
-    )
+    with gr.Row():
+        file_uploads = gr.File(
+            label="Upload Word Document Essays to Grade",
+            file_count="multiple",
+            file_types=[".docx"],
+            type="filepath",
+            scale=2
+        )
+        example_paper_upload = gr.File(
+            label="Upload Example PDF Paper (Optional)",
+            file_count="single",
+            file_types=[".pdf"],
+            type="filepath",
+            scale=1
+        )
 
     grade_button = gr.Button("🚀 Grade All Papers", variant="primary")
-
     gr.Markdown("---")
     gr.Markdown("## 📊 Grading Results")
-
     results_output = gr.Markdown(label="Formatted Grades")
-
     status_output = gr.Textbox(
         label="Runtime Status",
         lines=2,
@@ -564,9 +607,9 @@ with gr.Blocks(theme=gr.themes.Soft(), title="Nursing Essay Grader") as demo:
 
     grade_button.click(
         fn=grade_papers_concurrently,
-        inputs=[file_uploads, api_key_input],
+        inputs=[file_uploads, example_paper_upload, api_key_input],
         outputs=[results_output, status_output]
     )
 
 if __name__ == "__main__":
-    demo.launch(debug=True)
+    demo.launch(debug=True)