Update sherlock2.py

sherlock2.py

@@ -213,6 +213,12 @@ def search_internet(case_text):
 
     return internet_search_results
 
+# Function to display chat history with highlighted user input and chatbot response
+def display_chat_history():
+    for user_msg, bot_msg in st.session_state.chat_history:
+        st.info(f"**You:** {user_msg}")
+        st.success(f"**Sherlock:** {bot_msg}")
+
 def investigate():
     """Handles the case investigation process, including file upload, text extraction, embedding generation, 
     image processing, information analysis using Gemini models, web/Wikipedia search, and case report generation.
@@ -269,6 +275,16 @@ def investigate():
         st.header("Case Report")
         st.write(final_report.text)
 
+        user_query = st.text_input("Ask Sherlock:")
+        if user_query:
+            response = model.generate_content([sherlock_persona, sherlock_guidelines, user_query])
+            st.write(response.text)
+            st.session_state.chat_history.append((user_query, response.text))  # Add chat to history
+            display_chat_history()  # Display chat history
+
+        # Call the fixInputToBottom function after rendering the input text box
+        st.markdown("<script>fixInputToBottom();</script>", unsafe_allow_html=True)
+
     else:
         st.info("Please upload both case documents and images to proceed with the investigation.")
 
@@ -278,6 +294,12 @@ def investigate():
         if user_query:
             response = model.generate_content([sherlock_persona, sherlock_guidelines, user_query])
             st.write(response.text)
+            st.session_state.chat_history.append((user_query, response.text))  # Add chat to history
+            display_chat_history()  # Display chat history
+
+        # Call the fixInputToBottom function after rendering the input text box
+        st.markdown("<script>fixInputToBottom();</script>", unsafe_allow_html=True)
+
 def main():
     # --- Vintage Sherlock Holmes Theme ---
     st.set_page_config(page_title="AI Detective Sherlock Holmes", page_icon=":mag_right:")
@@ -295,7 +317,11 @@ def main():
     }
     .stTextInput > div > div > input {
         border: 1px solid #8b4513;
-        border-radius: 5px; 
+        border-radius: 5px;
+        padding: 10px;
+        font-size: 16px;
+        width: 100%;
+        box-sizing: border-box;
     }
     .stButton > button {
         background-color: #8b4513;
@@ -307,10 +333,33 @@ def main():
     """
     st.markdown(vintage_css, unsafe_allow_html=True)  # Apply custom CSS
 
+    # JavaScript function to fix the input text box to the bottom of the screen
+    st.markdown(
+        """
+        <script>
+            function fixInputToBottom() {
+                const inputContainer = document.querySelector('.stTextInput');
+                if (inputContainer) {
+                    inputContainer.style.position = 'fixed';
+                    inputContainer.style.bottom = '20px';
+                    inputContainer.style.left = '20px';
+                    inputContainer.style.right = '20px';
+                    inputContainer.style.zIndex = '999';
+                }
+            }
+            window.addEventListener('load', fixInputToBottom);
+        </script>
+        """,
+        unsafe_allow_html=True
+    )
+
     # Title and Header
     st.title("AI Detective Sherlock Holmes")
     st.header("_'Elementary, my dear Watson!'_")
 
+    # Display chat history
+    display_chat_history()
+
     # Add a sidebar for navigation
     st.sidebar.title("Navigation")
     options = ["Investigate Case", "Chat with Sherlock"]
@@ -325,6 +374,11 @@ def main():
         if user_query:
             response = model.generate_content([sherlock_persona, sherlock_guidelines, user_query])
             st.write(response.text)
+            st.session_state.chat_history.append((user_query, response.text))  # Add chat to history
+            display_chat_history()  # Display chat history
+
+        # Call the fixInputToBottom function after rendering the input text box
+        st.markdown("<script>fixInputToBottom();</script>", unsafe_allow_html=True)
 
 if __name__ == "__main__":
     main()