Update app.py

app.py

@@ -14,22 +14,21 @@ 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("**")
-    
-    # Initialize the formatted output with the bird name
-    formatted_output = f"**{sections[0].strip()}**\n"
-    
-    # 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"
-    
-    # Return the formatted output
-    return formatted_output.strip()
+  input_text = input_text.replace("!:", ":")
+
+  # Find all words before ': **' and replace with bold markdown
+  output_text = re.sub(r'(\w+)\s*:\s*\*\*', r'**\1**:', input_text)
+
+  # Replace double asterisks with double hashtags for remaining headings
+  output_text = output_text.replace("**", "##")
+
+  # Remove any extra whitespace at the beginning of lines
+  output_text = re.sub(r'^\s+', '', output_text, flags=re.MULTILINE)
+
+  # Add an extra newline after each heading (after the colon)
+  output_text = "## " + re.sub(r'(##\s*.*):', r'\1:\n\n', output_text)
+
+  return output_text
 
 
 @spaces.GPU