Update app.py

app.py

@@ -1,4 +1,5 @@
 import os
+import re
 import torch
 from threading import Thread
 from typing import Iterator
@@ -56,6 +57,18 @@ def make_prompt(entry):
     return  f"### Human: Don't repeat the assesments, limit to 500 words {entry} ### Assistant:"
     # f"TELL A STORY, RELATE TO COMPUTER SCIENCE, INCLUDE ASSESMENTS. MAKE IT REALISTIC AND AROUND 800 WORDS: {entry}"
 
+def process_text(text):
+    # Bold specific bracketed phrases
+    text = re.sub(r'\[(assessment.*?)]', r'**[\1]**', text, flags=re.IGNORECASE)
+    text = re.sub(r'\[(answer.*?)]', r'**[\1]**', text, flags=re.IGNORECASE)
+    
+    # Remove all other bracketed content
+    text = re.sub(r'\[.*?\]', '', text)  # Removes any text in square brackets
+    text = re.sub(r'\(.*?\)', '', text)  # Removes any text in round brackets
+    text = re.sub(r'\{.*?\}', '', text)  # Removes any text in curly brackets
+    
+    return text
+
 # Gradio Function
 @spaces.GPU
 def generate(
@@ -94,7 +107,8 @@ def generate(
 
     outputs = []
     for text in streamer:
-        outputs.append(text)
+        processed_text = process_text(text)
+        outputs.append(processed_text)
         yield "".join(outputs)
     final_story = "".join(outputs)
     try: