Update app.py

app.py

@@ -11,30 +11,26 @@ def classify_feedback(feedback_text):
     labels = ["Value", "Facilities", "Experience", "Functionality", "Quality"]
     result = classifier(feedback_text, labels, multi_label=True)
     
-    # Get the top two labels associated with the feedback and their scores
+    # Get the top two labels associated with the feedback
     top_labels = result["labels"][:2]
     scores = result["scores"][:2]
     
-    # Prepare the outputs to display both labels and their corresponding meters
-    outputs = []
-    for label, score in zip(top_labels, scores):
-        label_with_score = f"{label}: {score:.2f}"
-        outputs.append(gr.Label(label_with_score))
-        outputs.append(gr.Meter(value=score))
+    # Generate HTML content for displaying the scores as meters/progress bars
+    html_content = ""
+    for i in range(len(top_labels)):
+        score_percentage = scores[i] * 100  # Convert score to percentage
+        html_content += f"<div><b>{top_labels[i]}:</b> {scores[i]:.2f} <div style='background-color: #e0e0e0; border-radius: 10px;'><div style='height: 24px; width: {score_percentage}%; background-color: #76b900; border-radius: 10px;'></div></div></div>"
     
-    return outputs
+    return html_content
 
 # Create Gradio interface
 feedback_textbox = gr.Textbox(label="Enter your feedback:")
-feedback_output = [gr.Label(), gr.Meter(), gr.Label(), gr.Meter()]  # Output placeholders for two labels and meters
+feedback_output = gr.HTML(label="Top 2 Labels with Scores:")
 
-iface = gr.Interface(
+gr.Interface(
     fn=classify_feedback,
     inputs=feedback_textbox,
     outputs=feedback_output,
     title="Feedback Classifier",
-    description="Enter your feedback and get the top 2 associated labels with scores.",
-    layout="vertical"
-)
-
-iface.launch()
+    description="Enter your feedback and get the top 2 associated labels with scores."
+).launch()