Update app.py

app.py

@@ -90,20 +90,20 @@ def process_text(inputs: str, confidence_threshold: float):
                 if last_end < subtoken_start:
                     reconstructed_text += chunk[last_end:subtoken_start]
                 if word_data['prediction'] == 1:
-                    reconstructed_text += f"<span style='background-color: rgba(0, 255, 0); display: inline;'>{subtoken_text}</span>"
+                    reconstructed_text += f"<span style='background-color: rgba(0, 255, 0); display: inline;'>{subtoken_text.replace('$', '\\$')}</span>"
                 else:
-                    reconstructed_text += subtoken_text
+                    reconstructed_text += subtoken_text.replace('$', '\\$')
                 last_end = subtoken_end
 
-                df_data['Word'].append(subtoken_text)
+                df_data['Word'].append(subtoken_text.replace('$', '\\$'))
                 df_data['Prediction'].append(word_data['prediction'])
-                df_data['Confidence'].append(word_data['confidence'])
+                df_data['Confidence'].append(word_info[word_start]['confidence'])
                 df_data['Start'].append(subtoken_start + original_position_offset)
                 df_data['End'].append(subtoken_end + original_position_offset)
 
             original_position_offset += len(chunk) + 1
 
-        reconstructed_text += chunk[last_end:]
+        reconstructed_text += chunk[last_end:].replace('$', '\\$')
 
     df_tokens = pd.DataFrame(df_data)
     return reconstructed_text, df_tokens