Update sherlock2.py

sherlock2.py

@@ -329,6 +329,20 @@ def investigate():
                 st.header("Case Report")
                 st.write(final_report.text)
 
+def chat_with_sherlock():
+    """Handles the chat interaction with Sherlock Holmes."""
+    st.header("Consult with Sherlock")
+
+    # Display chat history
+    display_chat_history()
+
+    user_query = st.text_input("Ask Sherlock:", key="user_input")
+    if user_query:
+        chat_history_summary = summarize_chat_history(st.session_state.chat_history)
+        conversation_history = [sherlock_persona, sherlock_guidelines, chat_history_summary, f"Human: {user_query}"]
+        response = model.generate_content(conversation_history)
+        st.session_state.chat_history.append((user_query, response.text)) 
+        display_chat_history() 
 
 def main():
     # --- Vintage Sherlock Holmes Theme ---
@@ -337,7 +351,28 @@ def main():
     # Custom CSS for Styling
     vintage_css = """
     <style>
-    ... (CSS code remains the same) ...
+    body {
+        background-color: #d2b48c; /* Antique White */
+        color: #332200; /* Dark Brown */
+        font-family: 'Times New Roman', serif; 
+    }
+    h1, h2, h3 {
+        color: #8b4513; /* Saddle Brown */
+    }
+    .stTextInput > div > div > input {
+        border: 1px solid #8b4513;
+        border-radius: 5px;
+        padding: 10px;
+        font-size: 16px;
+        width: 100%;
+        box-sizing: border-box;
+    }
+    .stButton > button {
+        background-color: #8b4513;
+        color: white;
+        border: none;
+        border-radius: 5px;
+    }
     </style>
     """
     st.markdown(vintage_css, unsafe_allow_html=True)  # Apply custom CSS
@@ -346,33 +381,14 @@ def main():
     st.title("AI Detective Sherlock Holmes")
     st.header("_'Elementary, my dear Watson!'_")
 
-    # Initialize chat history
-    if 'chat_history' not in st.session_state:
-        st.session_state.chat_history = []
-
-    # 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}")
-
-    st.sidebar.title("Case Files")
-    st.sidebar.info("Upload your case files and let Sherlock analyze the evidence.")
-    
-    investigate()
-
-    # Chat with Sherlock
-    st.sidebar.title("Consult with Sherlock")
-    user_query = st.sidebar.text_input("Ask Sherlock:", key="user_input")
-    if user_query:
-        chat_history_summary = summarize_chat_history(st.session_state.chat_history)
-        conversation_history = [sherlock_persona, sherlock_guidelines, chat_history_summary, f"Human: {user_query}"]
-        response = model.generate_content(conversation_history)
-        st.session_state.chat_history.append((user_query, response.text)) 
-        display_chat_history()
+    pages = {
+        "Investigate a Case": investigate,
+        "Chat with Sherlock": chat_with_sherlock
+    }
 
-    # Call the fixInputToBottom function after rendering the input text box
-    st.markdown("<script>fixInputToBottom();</script>", unsafe_allow_html=True)
+    st.sidebar.title("Navigation")
+    page = st.sidebar.radio("Choose an action:", list(pages.keys()))
+    pages[page]()
 
 if __name__ == "__main__":
     main()