Update app.py

app.py

@@ -467,7 +467,21 @@ if not st.session_state.get("chat_started", False):
 
 
 
-
+def preprocess_text(text):
+    # Remove extra spaces
+    text = re.sub(r'\s+', ' ', text).strip()
+    
+    # Ensure proper spacing after punctuation
+    text = re.sub(r'([.,!?])(\S)', r'\1 \2', text)
+    
+    # Replace markdown bold and italic with HTML tags
+    text = re.sub(r'\*\*(.*?)\*\*', r'<strong>\1</strong>', text)
+    text = re.sub(r'\*(.*?)\*', r'<em>\1</em>', text)
+    
+    # Ensure proper line breaks
+    text = text.replace('\n', '<br>')
+    
+    return text
 prompt = st.chat_input("")
 if prompt :
 
@@ -492,19 +506,18 @@ if prompt :
             })
             full_response = output["output"]
 
-
-
+            processed_response = preprocess_text(full_response)
 
             # Display the response
+            st.markdown(processed_response, unsafe_allow_html=True)
 
-            st.write(full_response)
             follow_up_output = agent_executor.invoke({
                 "input": f"Extract 1-2 follow-up questions from the following text: {full_response}",
                 "chat_history": st.session_state.chat_history
             })
             follow_up_questions = follow_up_output["output"]
-            if follow_up_questions:
-                st.markdown(f"Follow-up questions:\n{follow_up_questions} \n")
+            st.session_state.follow_up_questions = preprocess_text(follow_up_questions)
+
 
 
         except Exception as e:
@@ -514,4 +527,12 @@ if prompt :
 
     # Add AI response to chat history
     st.session_state.chat_history.append({"role": "assistant", "content": full_response})
-    copy_to_clipboard(full_response)
+    copy_to_clipboard(full_response)
+
+
+
+if st.session_state.get("follow_up_questions"):
+    st.markdown("Follow-up questions:")
+    st.markdown(st.session_state.follow_up_questions, unsafe_allow_html=True)
+    # Clear follow-up questions for the next iteration
+    st.session_state.follow_up_questions = None