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():