Update gradio_interface.py

gradio_interface.py

@@ -40,16 +40,19 @@ def evaluate_text(input_text):
     # Additional classification report
     report = classification_report(ground_truth_labels_list, predicted_labels_list)
 
-    # Construct output
-    evaluation_results = {
-        "Accuracy": accuracy,
-        "Precision": precision,
-        "Recall": recall,
-        "F1-Score": f1,
-        "Classification Report": report
+    # Construct output dictionary
+    output_dict = {
+        "PredictedLabels": predicted_labels,
+        "EvaluationMetrics": {
+            "Accuracy": accuracy,
+            "Precision": precision,
+            "Recall": recall,
+            "F1-Score": f1,
+            "ClassificationReport": report
+        }
     }
     
-    return evaluation_results
+    return output_dict
 
 # Gradio Interface
 iface = gr.Interface(fn=evaluate_text, inputs="text", outputs="json", title="Text Evaluation", description="Enter your text")