Update app.py

app.py

@@ -11,28 +11,62 @@ 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:']
-    
-    # Split into title and content based on the first ":", handling extra whitespace
-    title, content = map(str.strip, input_text.split(":", 1))
-    # Bold the keywords 
+
+    bold_words = ['look:', 'cool fact!:', 'habitat:', 'food:', 'birdie behaviors:']
+
+    # Split into title and content, handle missing ":"
+    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 = content.replace(word, f'\n\n**{word}')
-    
-    # Construct the Markdown output with headings
-    formatted_output = f"**{title}**{content}" 
+        content = re.sub(rf"\b({word.lower()})\b", r"**\1**", content.lower()) 
 
+    # Construct Markdown output
+    formatted_output = f"**{title}**\n{content}" 
     return formatted_output.strip()
-
+    
 @spaces.GPU
 def infer_fin_pali(image, question):
     device = torch.device("cuda" if torch.cuda.is_available() else "cpu")