Update app.py

app.py

@@ -162,6 +162,9 @@ def add_paper_buttons_and_links():
 
 add_paper_buttons_and_links()
 
+
+
+# Process user input is a post processor algorithm which runs after document embedding vector DB play of GPT on context of documents..
 def process_user_input(user_question):
     response = st.session_state.conversation({'question': user_question})
     st.session_state.chat_history = response['chat_history']
@@ -178,19 +181,25 @@ def process_user_input(user_question):
         create_expanders_and_buttons(message.content)
 
 def create_expanders_and_buttons(content):
-    # Split the content into sections based on headers
-    sections = content.split("\n\n")
-    for section in sections:
-        lines = section.split("\n")
-        if len(lines) > 1 and ":" in lines[0]:  # Check for header and details
-            header = lines[0].replace(":", "").strip()
-            detail = " ".join(lines[1:]).strip()
-
+    # Split the content into paragraphs
+    paragraphs = content.split("\n\n")
+    for paragraph in paragraphs:
+        # Identify the header and detail in the paragraph
+        header, detail = extract_feature_and_detail(paragraph)
+        if header and detail:
             with st.expander(header, expanded=False):
                 if st.button(f"Explore {header}"):
                     expanded_outline = "Expand on the feature: " + detail
                     chat_with_model(expanded_outline, header)
 
+def extract_feature_and_detail(paragraph):
+    # Use regex to find the header and detail in the paragraph
+    match = re.match(r"(.*?):(.*)", paragraph)
+    if match:
+        header = match.group(1).strip()
+        detail = match.group(2).strip()
+        return header, detail
+    return None, None
 
 
 def process_user_input_old(user_question):