Update app.py

app.py

@@ -32,6 +32,26 @@ def bold_predicted_letters(input_string: str, output_string: str) -> str:
     
     return "".join(result)
 
+def diff_texts(string):
+    new_string = []
+
+    bold = False
+
+    for idx, letter in enumerate(string):
+
+        if letter == '*' and string[min(idx + 1, len(string)-1)] == '*' and bold == False:
+            bold = True
+
+        elif letter == '*' and string[min(idx + 1, len(string)-1)] == '*' and bold == True:
+            bold = False
+        if letter != '*':
+            if bold :
+                new_string.append((letter,'+'))
+            else:
+                new_string.append((letter, None))
+
+    return new_string
+    
 class model:
     def __init__(self):
         self.model = None
@@ -105,7 +125,8 @@ class model:
         formatted_predicted_sequence = formatted_predicted_sequence.replace("<cls>","")
         formatted_predicted_sequence = formatted_predicted_sequence.replace("<eos>","")
 
-        formatted_predicted_sequence = bold_predicted_letters(sequence_input, formatted_predicted_sequence)                                                      
+        formatted_predicted_sequence = bold_predicted_letters(sequence_input, formatted_predicted_sequence) 
+        formatted_predicted_sequence = diff_texts(formatted_predicted_sequence)
         return T.ToPILImage()(protein_image[0,0]), T.ToPILImage()(nucleus_image[0,0]), formatted_predicted_sequence
 
 base_class = model()
@@ -169,7 +190,11 @@ with gr.Blocks(theme='gradio/soft') as demo:
         gr.Markdown("Sequence predictions are show below.")
 
     with gr.Row().style(equal_height=True):
-        predicted_sequence = gr.Markdown(label='Predicted Sequence')
+        # predicted_sequence = gr.Markdown(label='Predicted Sequence')
+        predicted_sequence = gr.HighlightedText(
+        label="Predicted Sequence",
+        combine_adjacent=True,
+        show_legend=False).style(color_map={"+": "green"})
 
 
     with gr.Row():