Update app.py

app.py

@@ -72,10 +72,25 @@ def evaluate_model(labels, preds):
     cm = confusion_matrix(labels, preds)
     accuracy = accuracy_score(labels, preds)
     roc_score = roc_auc_score(labels, preds)
-    report = classification_report(labels, preds)
+    
+    # Generate classification report as a dictionary
+    report_dict = classification_report(labels, preds, output_dict=True)
+    
+    # Formatting the report as an HTML string for cleaner display
+    report_html = "<h4>Classification Report</h4>"
+    report_html += "<table><tr><th>Class</th><th>Precision</th><th>Recall</th><th>F1-Score</th></tr>"
+    
+    for key, value in report_dict.items():
+        if isinstance(value, dict):
+            report_html += f"<tr><td>{key}</td><td>{value['precision']:.2f}</td><td>{value['recall']:.2f}</td><td>{value['f1-score']:.2f}</td></tr>"
+    
+    report_html += "</table>"
+    
+    # Compute ROC Curve
     fpr, tpr, _ = roc_curve(labels, preds)
     roc_auc = auc(fpr, tpr)
 
+    # Plot Confusion Matrix and ROC Curve
     fig, (ax1, ax2) = plt.subplots(1, 2, figsize=(15, 6))
     
     ConfusionMatrixDisplay(confusion_matrix=cm, display_labels=LABELS).plot(cmap=plt.cm.Blues, ax=ax1)
@@ -92,7 +107,7 @@ def evaluate_model(labels, preds):
     
     plt.tight_layout()
 
-    return accuracy, roc_score, report, fig
+    return accuracy, roc_score, report_html, fig
 
 def load_url(url):
     try:
@@ -134,7 +149,7 @@ def create_gradio_interface():
                     gr.Markdown(f"### Results for {MODEL_NAME}")
                     output_acc = gr.Label(label="Accuracy")
                     output_roc = gr.Label(label="ROC Score")
-                    output_report = gr.Textbox(label="Classification Report", lines=10)
+                    output_report = gr.HTML(label="Classification Report")  # Changed to gr.HTML for formatted display
                     output_plots = gr.Plot(label="Confusion Matrix and ROC Curve")
 
         load_btn.click(load_url, in_url, [inp, message])
@@ -154,4 +169,4 @@ def create_gradio_interface():
 
 if __name__ == "__main__":
     app = create_gradio_interface()
-    app.launch(show_api=False, max_threads=24)
+    app.launch(show_api=False, max_threads=24)