Update app.py

app.py

@@ -15,11 +15,14 @@ def predict(text):
     for result in all_scores:
         label = result['label']
         score = result['score']
-        # Wrap output in HTML for green background and black text
-        response += f"<span style='background-color:green; color:black;'>{label}: {score:.3f}</span><br>"
+        # Apply special styling only for "Politeness", "Sentiment", "Formality"
+        if label in ["Politeness", "Sentiment", "Formality"]:
+            response += f"<span style='background-color:green; color:black;'>{label}: {score:.3f}</span><br>"
+        else:
+            response += f"{label}: {score:.3f}<br>"
     
     if not re.match(greeting_pattern, text, re.IGNORECASE):
-        greeting_feedback = "<span style='background-color:green; color:black;'>\n- Heilsaðu dóninn þinn\n</span>"
+        greeting_feedback = "Heilsaðu dóninn þinn<br>"
 
     response += greeting_feedback
     
@@ -35,7 +38,7 @@ textbox = gr.Textbox()
 demo = gr.Interface(
     fn=predict,
     inputs=gr.TextArea(label="Enter text here:"),
-    outputs= gr.HTML(label="Leiðrétt"),
+    outputs=gr.HTML(label="Leiðrétt"),
     description=description_html,
     examples=[
         ["Það voru vitni að árásinni sem tilkynntu málið til lögreglu sem kom skjótt á vettvang."],
@@ -44,7 +47,7 @@ demo = gr.Interface(
         ["Hver á þenan bússtað? já eða nei."],
         ["Hafi þau svo látið gólfið þorna vel og síðan flotað það til lagfæringar eftir motturnar."],
     ],
-    theme = gr.themes.Default(primary_hue="red", secondary_hue="pink")
+    theme=gr.themes.Default(primary_hue="red", secondary_hue="pink")
 )
 
-demo.launch()
+demo.launch()