Update app.py

samadpls · Aug 20, 2024 · 474d5c4 · 474d5c4
1 parent e6d9526
commit 474d5c4
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/app.py b/src/app.py
@@ -125,7 +125,11 @@ def main():
                     with st.chat_message("assistant"):
                         with st.spinner("Generating..."):
                             response = llm_chain.invoke({"input": prompt})
-                            st.markdown(response)
+                            index = response.find("```")
+                            if index != -1:
+                                st.markdown(response[index:])
+                            else:
+                                st.markdown(response)
                             message = {
                                 "role": "assistant",
                                 "content": response,