Update app.py

app.py

@@ -25,21 +25,20 @@ def convert_to_markdown(input_text):
         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 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')
-    #     content = content.replace(f'** ', f' ')
+    # 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')
+        content = content.replace(f'** ', f' ')
     
-    # # Construct the Markdown output with headings
-    # formatted_output = f"**{title}{content}" 
-    input_text = """**ABBOTT'S BABBLER (Malacocincla abbotti)** \n\n **Look:** \n  Robin-sized detective! This bird has a sandy-brown body with rusty flanks, a short tail, and a heavy hooked bill. Check for a pale gray eyebrow in Southeast Asia. \n\n **Cool Fact!:** \n  Works in pairs, hopping on the forest floor like little detectives searching for clues (their prey!). \n\n **Habitat:** \n  Prefers the shady undergrowth of permanent forests, especially near streams and tangled vegetation. \n\n **Food:** \n  Not picky eaters! They enjoy a varied menu of insects, worms, and even seeds. \n\n **Birdie Behaviors:** \n  Secretive by nature, but listen for their surprising calls – a mix of harsh churrs and melodic whistles."""
+    # Construct the Markdown output with headings
+    formatted_output = f"**{title}{content}" 
 
 
-    return input_text
+    return formatted_output.strip()
 
 @spaces.GPU
 def infer_fin_pali(image, question):