Update app.py

app.py

@@ -83,7 +83,7 @@ def get_word_probabilities(text):
     if current_word and current_probs:
         current_prob = sum(current_probs) / len(current_probs)
         word_probs.append(current_prob)
-            color = ("green" if current_prob < 0.25 else "yellow" if current_prob < 0.5 else "orange" if current_prob < 0.75 else "red")
+        color = ("green" if current_prob < 0.25 else "yellow" if current_prob < 0.5 else "orange" if current_prob < 0.75 else "red")
         word_colors.append(color)
     word_probs = [float(p) for p in word_probs]
     return word_probs, word_colors
@@ -104,7 +104,8 @@ def infer_and_log(text_input):
         except Exception as e:
             print(f"Error uploading log: {e}")
     tokens = text_input.split()
-    formatted_output = " ".join(f'<span style="color:{color}">{token}</span>' for token, color in zip(tokens, word_colors))
+    
+    formatted_output = " ".join(f'<span style="background-color:#1F1F23; color:{color}">{token}</span>' for token, color in zip(tokens, word_colors))
     return formatted_output
 
 def clear_fields():