Update app.py

app.py

@@ -63,12 +63,13 @@ if st.button("Generate", use_container_width=True):
             try:
                 # Extracting generated response
                 assistant_message = result["choices"][0]["message"]["content"]
-
-                # Remove `**` from the response
-                cleaned_message = assistant_message.replace("**", "")
                 
-                # Displaying output in a code block
-                st.code(assistant_message, language="text")
+                # Remove Markdown-like formatting characters
+                import re
+                cleaned_message = re.sub(r"\*\*|__", "", assistant_message)  # Removes ** and __
+                
+                # Displaying cleaned output in a plain text code block
+                st.code(cleaned_message, language="text")
             except KeyError as e:
                 st.error(f"Unexpected response format: {e}")
     else: