Update app.py

app.py

@@ -2054,27 +2054,29 @@ user_name = extract_name(st.session_state["email"])
 
 
 
-# def clean_and_format_markdown(raw_text):
-#     """
-#     Cleans and formats Markdown text to ensure URLs are properly encoded 
-#     and handles issues with line breaks or improperly formatted Markdown.
-#     """
-#     # Regular expression to find Markdown links [text](url)
-#     pattern = r'\[([^\]]+)\]\(([^)]+)\)'
+def clean_and_format_markdown(raw_text):
+    """
+    Cleans and formats Markdown text to ensure URLs are properly encoded 
+    and handles issues with line breaks or improperly formatted Markdown.
+    """
+    # Regular expression to find Markdown links [text](url)
+    pattern = r'\[([^\]]+)\]\(([^)]+)\)'
+
+    def encode_url(match):
+        text = match.group(1)
+        url = match.group(2).strip()  # Remove leading/trailing spaces
+        encoded_url = quote(url, safe=':/')  # Encode the URL while keeping : and /
+        corrected_url = encoded_url.replace("+", "%20")
 
-#     def encode_url(match):
-#         text = match.group(1)
-#         url = match.group(2).strip()  # Remove leading/trailing spaces
-#         encoded_url = quote(url, safe=':/')  # Encode the URL while keeping : and /
-#         return f"[{text}]({encoded_url})"
+        return f"[{text}]({corrected_url})"
 
-#     # Fix Markdown links dynamically
-#     formatted_text = re.sub(pattern, encode_url, raw_text)
+    # Fix Markdown links dynamically
+    formatted_text = re.sub(pattern, corrected_url, raw_text)
 
-#     # Replace single newlines with spaces to avoid breaking Markdown rendering
-#     formatted_text = re.sub(r'(?<!\n)\n(?!\n)', ' ', formatted_text)
+    # Replace single newlines with spaces to avoid breaking Markdown rendering
+    formatted_text = re.sub(r'(?<!\n)\n(?!\n)', ' ', formatted_text)
 
-#     return formatted_text
+    return formatted_text
 
 prompt = st.chat_input("")
 global combined_text
@@ -2222,8 +2224,8 @@ def handle_prompt(prompt):
                     })
                     full_response = output["output"]
 
-                    formatted_text = clean_text(full_response)
-                    #formatted_text = clean_and_format_markdown(cleaned_text)
+                    cleaned_text = clean_text(full_response)
+                    formatted_text = clean_and_format_markdown(cleaned_text)
 
                     trust_tip, suggestion = get_trust_tip_and_suggestion()
                     combined_text = f"{formatted_text}\n\n---\n\n**Trust Tip**: {trust_tip}\n\n**Suggestion**: {suggestion}"