Update app.py

app.py

@@ -15,21 +15,28 @@ bnb_config = BitsAndBytesConfig(load_in_8bit=True)
 
 
 def convert_to_markdown(input_text):
-    # Split the input text into sections based on the '**' delimiter
-    sections = input_text.split("**")
+    """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.
+    """
     
-    # Initialize the formatted output with the bird name
-    formatted_output = f"**{sections[0].strip()}**\n"
+    bold_words = ['Look:', 'Cool Fact!:', 'Habitat:', 'Food:', 'Birdie Behaviors:']
     
-    # Process each section to format it
-    for i in range(1, len(sections), 2):
-        if i + 1 < len(sections):
-            # Use '##' for subheadings and clean up the text
-            header = sections[i].strip() + "** "
-            content = sections[i + 1].strip()
-            formatted_output += f"\n**{header}{content}\n"
+    # Split into title and content based on the first ":", handling extra whitespace
+    title, content = map(str.strip, input_text.split(":", 1))
+
+    # Bold the keywords 
+    for word in bold_words:
+        content = content.replace(word, f'\n**{word}**\n')
     
-    # Return the formatted output
+    # Construct the Markdown output with headings
+    formatted_output = f"** {title} **\n{content}" 
+
     return formatted_output.strip()
 
 @spaces.GPU