Update app.py

app.py

@@ -1,12 +1,24 @@
 import gradio as gr
 from transformers import pipeline, PipelineException
+import fitz  # PyMuPDF
 
 # Load a summarization model from Hugging Face
 summarizer = pipeline("summarization")
 
-def evaluate_text_against_rubric(rubric, text):
+def extract_text_from_pdf(pdf_file):
+    text = ""
+    try:
+        document = fitz.open(stream=pdf_file.read(), filetype="pdf")
+        for page_num in range(len(document)):
+            page = document.load_page(page_num)
+            text += page.get_text()
+    except Exception as e:
+        return str(e)
+    return text
+
+def evaluate_text_against_rubric(rubric_text, text):
     # Split rubric into criteria
-    criteria = [criterion.strip() for criterion in rubric.split('\n') if criterion.strip()]
+    criteria = [criterion.strip() for criterion in rubric_text.split('\n') if criterion.strip()]
     
     if not criteria:
         return "No valid criteria found in the rubric."
@@ -41,7 +53,19 @@ def evaluate_text_against_rubric(rubric, text):
     
     return evaluations
 
-def evaluate(rubric, text):
+def evaluate(rubric_pdf, rubric_text, text):
+    rubric = ""
+    if rubric_pdf is not None:
+        rubric = extract_text_from_pdf(rubric_pdf)
+    elif rubric_text:
+        rubric = rubric_text
+    
+    if not rubric:
+        return "No rubric provided."
+    
+    if not text:
+        return "No text provided for evaluation."
+    
     evaluation = evaluate_text_against_rubric(rubric, text)
     
     if isinstance(evaluation, str):  # If it's an error message
@@ -58,17 +82,24 @@ def evaluate(rubric, text):
     return evaluation_text
 
 # Create Gradio interface
-interface = gr.Interface(
-    fn=evaluate,
-    inputs=[
-        gr.Textbox(lines=10, placeholder="Enter your rubric here..."),
-        gr.Textbox(lines=10, placeholder="Paste the text to be evaluated here...")
-    ],
-    outputs="text",
-    title="Text Evaluator",
-    description="Upload or enter a rubric and paste text for evaluation. The system will grade the text against the rubric and provide feedback."
-)
+with gr.Blocks() as interface:
+    gr.Markdown("# PDF Text Evaluator")
+    gr.Markdown("Upload a rubric as a PDF or paste the rubric text, then paste text for evaluation.")
+    
+    rubric_pdf_input = gr.File(label="Upload Rubric PDF (optional)", type="file")
+    rubric_text_input = gr.Textbox(lines=10, placeholder="Or enter your rubric text here...", label="Rubric Text (optional)")
+    text_input = gr.Textbox(lines=10, placeholder="Paste the text to be evaluated here...", label="Text to Evaluate")
+    
+    evaluate_button = gr.Button("Evaluate")
+    
+    output = gr.Textbox(label="Evaluation Results")
+    
+    def evaluate_button_clicked(rubric_pdf, rubric_text, text):
+        return evaluate(rubric_pdf, rubric_text, text)
+    
+    evaluate_button.click(evaluate_button_clicked, inputs=[rubric_pdf_input, rubric_text_input, text_input], outputs=output)
 
 # Launch the interface
 interface.launch()
 
+