Update app.py

app.py

@@ -34,45 +34,44 @@ Rules:
 6. Do not give marks outside the markscheme.
 Output must be a structured grading report with reasoning.
 """
-
-# ---------- CORE FUNCTION ----------
 def grade_student(qp_file, ms_file, ans_file):
     try:
-        # Upload files to Gemini
-        qp_uploaded = genai.upload_file(path=qp_file.name, display_name="Question Paper")
-        ms_uploaded = genai.upload_file(path=ms_file.name, display_name="Marking Scheme")
-        ans_uploaded = genai.upload_file(path=ans_file.name, display_name="Answer Sheet")
+        # Get file paths
+        qp_path = qp_file.name
+        ms_path = ms_file.name
+        ans_path = ans_file.name
+
+        # Upload to Gemini
+        qp_uploaded = genai.upload_file(path=qp_path, display_name="Question Paper")
+        ms_uploaded = genai.upload_file(path=ms_path, display_name="Marking Scheme")
+        ans_uploaded = genai.upload_file(path=ans_path, display_name="Answer Sheet")
 
-        # Step 1: Transcription
         model = genai.GenerativeModel("gemini-2.5-pro", generation_config={"temperature": 0})
+
         transcription_resp = model.generate_content([TRANSCRIPTION_PROMPT, ans_uploaded])
         transcription = getattr(transcription_resp, "text", None) or transcription_resp.candidates[0].content.parts[0].text
 
-        # Step 2: Grading
         grading_resp = model.generate_content([GRADING_PROMPT, qp_uploaded, ms_uploaded, transcription])
         grading = getattr(grading_resp, "text", None) or grading_resp.candidates[0].content.parts[0].text
 
         return transcription, grading
-
     except Exception as e:
-        return "❌ Error during processing: " + str(e), ""
-
+        return f"❌ Error during processing: {e}", ""
 
-# ---------- GRADIO UI ----------
 demo = gr.Interface(
     fn=grade_student,
     inputs=[
-        gr.File(label="Upload Question Paper (PDF)", type="file"),
-        gr.File(label="Upload Mark Scheme (PDF)", type="file"),
-        gr.File(label="Upload Student Answer Sheet (PDF)", type="file")
+        gr.File(label="Upload Question Paper (PDF)", type="filepath"),
+        gr.File(label="Upload Mark Scheme (PDF)", type="filepath"),
+        gr.File(label="Upload Student Answer Sheet (PDF)", type="filepath"),
     ],
     outputs=[
         gr.Markdown(label="Student Transcription"),
         gr.Textbox(label="Grading Report")
     ],
     title="📘 AI Teacher Assistant",
-    description="Upload a Question Paper, Mark Scheme, and Student Answer Sheet. The app transcribes handwritten answers into Markdown/LaTeX and grades them using the official scheme."
+    description="Upload a Question Paper, Mark Scheme, and Student Answer Sheet. The app transcribes and grades answers."
 )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()