Fix Classifier Score to Label Mappings in Gradio Interface

app.py

@@ -47,20 +47,11 @@ def classify_steam_review(input_text):
 
 
 # Provides a textual representation of the classification result
-def get_steam_review_classification_result_text(label_1, score_1, label_2, score_2):
-    # Maps label values to constructiveness
-    def label_to_constructiveness(label):
-        return "Constructive" if label == "LABEL_1" else "Not Constructive"
-
-    # Formats the output in a readable format
-    def format_output(label, score, emoji):
-        return f'{label_to_constructiveness(label)} with a score of {score}. {emoji}'
-
-    # Determines the label and score with the highest score
-    if score_1 >= score_2:
-        return format_output(label_1, score_1, "👍🏻")
+def get_steam_review_classification_result_text(label_1, score_1):
+    if label_1 == "LABEL_1":
+        return f"Constructive with a score of {score_1}. 👍🏻"
     else:
-        return format_output(label_2, score_2, "👎🏻")
+        return f"Not Constructive with a score of {score_1}. 👎🏻"
 
 
 # Examples Steam Reviews to display in the Gradio Interface using the "examples" parameter
@@ -74,7 +65,7 @@ examples = [
 # HTML article to display in the Gradio Interface using the "article" parameter
 article = (
     """
-    *Format your input as follows for the best results: **Review**: {review_text}, **Playtime**: {author_playtime_at_review}, **Voted Up**: {voted_up}, **Upvotes**: {upvotes}, **Votes Funny**: {votes_funny}.*
+    Format your input as follows for the best results: ***Review**: {review_text}, **Playtime**: {author_playtime_at_review}, **Voted Up**: {voted_up}, **Upvotes**: {upvotes}, **Votes Funny**: {votes_funny}.*
     """
 )
 
@@ -124,14 +115,18 @@ with gr.Blocks() as steam_reviews_classifier_block:
     # Function to run when the Submit Button is clicked (Passes the input text to the classifier and displays the output text)
     def on_submit_click(input_text):
         classification_result = classify_steam_review(input_text)
-        classification_result_text = get_steam_review_classification_result_text(
-            label_1=classification_result["label_1"],
-            score_1=classification_result["score_1"],
-            label_2=classification_result["label_2"],
-            score_2=classification_result["score_2"]
-        )
-        output_text = classification_result_text
-        constructive, not_constructive = str(classification_result["score_1"]), str(classification_result["score_2"])
+        label_1 = classification_result["label_1"]
+        score_1 = classification_result["score_1"]
+        score_2 = classification_result["score_2"]
+
+        output_text = get_steam_review_classification_result_text(label_1, score_1)
+
+        # Maps the scores to the constructive and not constructive labels in the Gradio Interface
+        if label_1 == "LABEL_1":
+            constructive, not_constructive = score_1, score_2
+        else:
+            constructive, not_constructive = score_2, score_1
+
         return output_text, constructive, not_constructive
 
 
@@ -139,8 +134,8 @@ with gr.Blocks() as steam_reviews_classifier_block:
     submit_button.click(
         fn=on_submit_click,
         inputs=input_textbox,
-        outputs=[output_textbox, not_constructive_label, constructive_label]
+        outputs=[output_textbox, constructive_label, not_constructive_label]
     )
 
 # Launches the Gradio Blocks Interface
-steam_reviews_classifier_block.launch()
+steam_reviews_classifier_block.launch()