updated flow

gradio_app.py

@@ -0,0 +1,36 @@
+import gradio as gr
+from main import flow
+
+
+with gr.Blocks(theme= "JohnSmith9982/small_and_pretty") as demo:
+    
+    gr.HTML(
+    """
+    <h1 style="font-size: 3.5em; text-align: center; color: #02c160; font-weight: bold;">AI Exam Checker</h1>
+    """
+            )
+    
+    with gr.Row():
+        with gr.Column():
+            student_pdf_path = gr.File(label="Student's PDF Response")
+        with gr.Column():
+            standard_pdf_path = gr.File(label="Standard Answer Key PDF")
+
+    with gr.Row():
+        submit_btn = gr.Button("Evaluate Exam")
+        
+    with gr.Row():
+        output = gr.File(label="Output PDF", interactive=False)
+
+
+    def evaluate_exam(student_pdf_path, standard_pdf_path):
+
+        save_dict = True
+        student_dict_path = "student_response.json"
+        standard_dict_path = "standard_key.json"
+        output_pdf_path = "output_report.pdf"
+
+        return flow(student_pdf_path, standard_pdf_path, save_dict, student_dict_path, standard_dict_path, output_pdf_path)
+
+    submit_btn.click(evaluate_exam, inputs=[student_pdf_path, standard_pdf_path], outputs=output)
+demo.launch(share=True, pwa=True, debug=True)

main.py

@@ -15,26 +15,18 @@ def flow(student_pdf_path: str,
 
     # create new interim_files folder if multiple instances of the flow are run
 
-    # Transcribe the student's PDF response
-    print("Transcribing the student's response")
-    student_resp = transcribe_pdf(student_pdf_path, "interim_files_student", save_dict, student_dict_path)
-    print("--------------------------------------------")
     # Transcribe the student's PDF response
     print("Transcribing the student's response")
     student_resp = transcribe_pdf(student_pdf_path, "interim_files_student", save_dict, student_dict_path)
     print("--------------------------------------------")
 
-    # Transcribe the standard answer key PDF
-    print("Transcribing the standard answer key")
-    standard_key = transcribe_pdf(standard_pdf_path, "interim_files_standard", save_dict, standard_dict_path)
-    print("--------------------------------------------")
     # Transcribe the standard answer key PDF
     print("Transcribing the standard answer key")
     standard_key = transcribe_pdf(standard_pdf_path, "interim_files_standard", save_dict, standard_dict_path)
     print("--------------------------------------------")
 
-    # comment if running full flow
-    # when testing the evaluator only
+    # # comment if running full flow
+    # # when testing the evaluator only
     # student_resp = json.load(open(student_dict_path, "r"))
     # standard_key = json.load(open(standard_dict_path, "r"))
 
@@ -67,7 +59,7 @@ if __name__ == "__main__":
     
     student_pdf_path = "examples/ca inter chapter - amalgamation student answer.pdf"
     standard_pdf_path = "examples/ca inter chapter - amalgamation.pdf"
-    
+
     save_dict = True
     student_dict_path = "student_response.json"
     standard_dict_path = "standard_key.json"