Update app.py

app.py

@@ -2,6 +2,8 @@ import streamlit as st
 import os
 import requests
 from dotenv import load_dotenv  # Only needed if using a .env file
+import re  # To help clean up leading whitespace
+
 
 # Langchain and HuggingFace
 from langchain.vectorstores import Chroma
@@ -68,36 +70,36 @@ if "messages" not in st.session_state:
     st.session_state.messages = [{"role": "assistant", "content": "How may I help you today?"}]
 
 # Function for generating response using the last three conversations
-def generate_response(prompt_input):
-    # Initialize result
-    result = ''
-    
-    # Prepare conversation history: get the last 3 user and assistant messages
-    conversation_history = ""
-    recent_messages = st.session_state.messages[-3:]  # Last 3 user and assistant exchanges (each exchange is 2 messages)
-
-    for message in recent_messages:
-        conversation_history += f"{message['role']}: {message['content']}\n"
-    
-    # Append the current user prompt to the conversation history
-    conversation_history += f"user: {prompt_input}\n"
-
-    # Invoke chain with the truncated conversation history
-    res = st.session_state.chain.invoke(conversation_history)
-    
-    # Process response (as in the original code)
-    if res['result'].startswith('According to the provided context, '):
-        res['result'] = res['result'][35:]
-        res['result'] = res['result'][0].upper() + res['result'][1:]
-    elif res['result'].startswith('Based on the provided context, '):
-        res['result'] = res['result'][31:]
-        res['result'] = res['result'][0].upper() + res['result'][1:]
-    elif res['result'].startswith('According to the provided text, '):
-        res['result'] = res['result'][34:]
-        res['result'] = res['result'][0].upper() + res['result'][1:]
-    elif res['result'].startswith('According to the context, '):
-        res['result'] = res['result'][26:]
-        res['result'] = res['result'][0].upper() + res['result'][1:]
+# def generate_response(prompt_input):
+#     # Initialize result
+#     result = ''
+    
+#     # Prepare conversation history: get the last 3 user and assistant messages
+#     conversation_history = ""
+#     recent_messages = st.session_state.messages[-3:]  # Last 3 user and assistant exchanges (each exchange is 2 messages)
+
+#     for message in recent_messages:
+#         conversation_history += f"{message['role']}: {message['content']}\n"
+    
+#     # Append the current user prompt to the conversation history
+#     conversation_history += f"user: {prompt_input}\n"
+
+#     # Invoke chain with the truncated conversation history
+#     res = st.session_state.chain.invoke(conversation_history)
+    
+#     # Process response (as in the original code)
+#     if res['result'].startswith('According to the provided context, '):
+#         res['result'] = res['result'][35:]
+#         res['result'] = res['result'][0].upper() + res['result'][1:]
+#     elif res['result'].startswith('Based on the provided context, '):
+#         res['result'] = res['result'][31:]
+#         res['result'] = res['result'][0].upper() + res['result'][1:]
+#     elif res['result'].startswith('According to the provided text, '):
+#         res['result'] = res['result'][34:]
+#         res['result'] = res['result'][0].upper() + res['result'][1:]
+#     elif res['result'].startswith('According to the context, '):
+#         res['result'] = res['result'][26:]
+#         res['result'] = res['result'][0].upper() + res['result'][1:]
         
     
     # result += res['result']
@@ -157,6 +159,7 @@ def generate_response(prompt_input):
     # return result_text
 
     # return res['result']
+
 def generate_response(prompt_input):
     # Retrieve vector database context using ONLY the current user input
     retriever = st.session_state.chain.retriever
@@ -182,14 +185,19 @@ def generate_response(prompt_input):
     
     # Process the response text
     result_text = res['result']
+    
+    # Clean up prefixing phrases and capitalize the first letter
     if result_text.startswith('According to the provided context, '):
-        result_text = result_text[35:].capitalize()
+        result_text = result_text[35:].strip()
     elif result_text.startswith('Based on the provided context, '):
-        result_text = result_text[31:].capitalize()
+        result_text = result_text[31:].strip()
     elif result_text.startswith('According to the provided text, '):
-        result_text = result_text[34:].capitalize()
+        result_text = result_text[34:].strip()
     elif result_text.startswith('According to the context, '):
-        result_text = result_text[26:].capitalize()
+        result_text = result_text[26:].strip()
+    
+    # Ensure the first letter is uppercase
+    result_text = result_text[0].upper() + result_text[1:] if result_text else result_text
 
     # Extract and format sources (if available)
     sources = []
@@ -207,8 +215,6 @@ def generate_response(prompt_input):
     
     return result_text
 
-
-
 # Display chat messages
 for message in st.session_state.messages:
     with st.chat_message(message["role"]):