Update app.py

app.py

@@ -253,15 +253,15 @@ async def grade_papers_concurrently(
             example_paper_file_obj = genai.upload_file(path=example_paper_file.name, display_name=os.path.basename(example_paper_file.name))
             
         file_paths = [file.name for file in files]
-        total_files = len(file_paths)
-
+        
         with ThreadPoolExecutor(max_workers=1) as executor:
             loop = asyncio.get_event_loop()
             tasks = [loop.run_in_executor(executor, process_single_file, fp, grader, example_paper_file_obj) for fp in file_paths]
+            # Use tqdm for progress tracking in the console/logs
             results = [await f for f in asyncio.as_completed(tasks)]
-            progress(1)
+            progress(1) # Mark progress as complete
 
-        # This section is updated to format the new, verbose JSON structure
+        # --- THIS SECTION IS UPDATED TO CALCULATE THE SCORE ---
         output_markdown = ""
         successful_grades = [res for res in results if res.success]
         failed_grades = [res for res in results if not res.success]
@@ -269,13 +269,19 @@ async def grade_papers_concurrently(
         for result in successful_grades:
             response_data = result.raw_response
             output_markdown += f"### ✅ Grade for: **{result.file_name}**\n"
-            output_markdown += f"**Final Score:** {response_data.get('finalScore', 'N/A')}/100\n\n"
+
+            # Calculate the score from the breakdown instead of trusting the AI's sum
+            breakdown = response_data.get('gradingBreakdown', [])
+            calculated_score = sum(item.get('score', 0) for item in breakdown)
+
+            output_markdown += f"**Final Score:** {calculated_score}/100\n\n"
             output_markdown += "**Detailed Grading Breakdown:**\n"
-            for item in response_data.get('gradingBreakdown', []):
+            for item in breakdown:
                 output_markdown += f"- **{item.get('criterion', 'N/A')}**: {item.get('score', 'N/A')} / {item.get('maxScore', 'N/A')}\n"
                 output_markdown += f"  - *Justification: {item.get('justification', 'No justification provided.')}*\n"
             output_markdown += f"\n**Summary:** {response_data.get('summary', 'No summary provided.')}\n"
             output_markdown += "---\n"
+        # --- END OF UPDATED SECTION ---
 
         if failed_grades:
             output_markdown += "### ❌ Failed Papers\n"