Update app.py

app.py

@@ -11,61 +11,33 @@ model_id = "selamw/BirdWatcher"
 bnb_config = BitsAndBytesConfig(load_in_8bit=True)
 
 
-# def convert_to_markdown(input_text):
-#     """Converts bird information text to Markdown format, 
-#        making specific keywords bold and adding headings.
-#     Args:
-#         input_text (str): The input text containing bird information.
-#     Returns:
-#         str: The formatted Markdown text.
-#     """
-    
-#     bold_words = ['Look:', 'Cool Fact!:', 'Habitat:', 'Food:', 'Birdie Behaviors:']
-
-#     # Split into title and content based on the first ":", handling extra whitespace
-#     if len(input_text.split(":", 1)) > 1:
-#         title, content = map(str.strip, input_text.split(":", 1))
-        
-#         # Bold the keywords 
-#         for word in bold_words:
-#             content = content.replace(word, f'\n\n**{word}')
-        
-#         # Construct the Markdown output with headings
-#         formatted_output = f"**{title}**{content}" 
-#     else:
-#         formatted_output = input_text
-#     return formatted_output.strip()
-
-import re
-
 def convert_to_markdown(input_text):
-    """
-    Converts bird information text to Markdown format, 
-    making specific keywords bold and adding headings.
-
+    """Converts bird information text to Markdown format, 
+       making specific keywords bold and adding headings.
     Args:
         input_text (str): The input text containing bird information.
-
     Returns:
         str: The formatted Markdown text.
     """
+    
+    bold_words = ['Look:', 'Cool Fact!:', 'Habitat:', 'Food:', 'Birdie Behaviors:']
 
-    bold_words = ['look:', 'cool fact!:', 'habitat:', 'food:', 'birdie behaviors:']
-
-    # Split into title and content, handle missing ":"
+    # Split into title and content based on the first ":", handling extra whitespace
     if ":" in input_text:
         title, content = map(str.strip, input_text.split(":", 1))
     else:
         title = input_text
         content = ""
 
-    # Bold the keywords (case-insensitive, word boundaries)
-    for word in bold_words:
-        content = re.sub(rf"\b({word.lower()})\b", r"**\1**", content.lower()) 
+        # Bold the keywords 
+        for word in bold_words:
+            content = content.replace(word, f'\n\n**{word}')
+        
+        # Construct the Markdown output with headings
+        formatted_output = f"**{title}**{content}" 
 
-    # Construct Markdown output
-    formatted_output = f"**{title}**\n{content}" 
     return formatted_output.strip()
+
     
 @spaces.GPU
 def infer_fin_pali(image, question):