Update app.py

app.py

@@ -1,17 +1,22 @@
 import gradio as gr
-from transformers import pipeline
+from transformers import pipeline, PipelineException
 
 # Load a summarization model from Hugging Face
 summarizer = pipeline("summarization")
 
 def evaluate_text_against_rubric(rubric, text):
     # Split rubric into criteria
-    criteria = rubric.split('\n')
+    criteria = [criterion.strip() for criterion in rubric.split('\n') if criterion.strip()]
+    
+    if not criteria:
+        return "No valid criteria found in the rubric."
+    
+    if not text:
+        return "No text provided for evaluation."
     
-    # Use summarizer for each criterion as a dummy placeholder
     evaluations = {}
     for i, criterion in enumerate(criteria):
-        if criterion.strip():
+        try:
             summary = summarizer(text, max_length=50, min_length=25, do_sample=False)[0]['summary_text']
             evaluations[f'Criteria {i+1}'] = {
                 "Criterion": criterion,
@@ -19,10 +24,29 @@ def evaluate_text_against_rubric(rubric, text):
                 "Comment": f"Evaluation based on criterion: {criterion}",
                 "Example": summary
             }
+        except PipelineException as e:
+            evaluations[f'Criteria {i+1}'] = {
+                "Criterion": criterion,
+                "Score": 0,
+                "Comment": f"Error during evaluation: {str(e)}",
+                "Example": ""
+            }
+        except Exception as e:
+            evaluations[f'Criteria {i+1}'] = {
+                "Criterion": criterion,
+                "Score": 0,
+                "Comment": f"An unexpected error occurred: {str(e)}",
+                "Example": ""
+            }
+    
     return evaluations
 
 def evaluate(rubric, text):
     evaluation = evaluate_text_against_rubric(rubric, text)
+    
+    if isinstance(evaluation, str):  # If it's an error message
+        return evaluation
+    
     evaluation_text = ""
     for criterion, details in evaluation.items():
         evaluation_text += f"{criterion}:\n"
@@ -30,6 +54,7 @@ def evaluate(rubric, text):
         evaluation_text += f"  Score: {details['Score']}\n"
         evaluation_text += f"  Comment: {details['Comment']}\n"
         evaluation_text += f"  Example: {details['Example']}\n\n"
+    
     return evaluation_text
 
 # Create Gradio interface
@@ -46,3 +71,4 @@ interface = gr.Interface(
 
 # Launch the interface
 interface.launch()
+