Update sherlock2.py

sherlock2.py

@@ -278,39 +278,6 @@ def investigate():
         if user_query:
             response = model.generate_content([sherlock_persona, sherlock_guidelines, user_query])
             st.write(response.text)
-
-def chat_with_sherlock():
-    """Handles the chat interaction with Sherlock Holmes."""
-    # Initialize chat history if it doesn't exist
-    if "chat_history" not in st.session_state:
-        st.session_state["chat_history"] = []
-
-    # Create input and output containers
-    input_container = st.container()
-    output_container = st.container()
-
-    with input_container:
-        user_input = st.text_input("Ask Sherlock:", key="user_input")
-        submit_button = st.button("Ask Sherlock", key="submit_button")
-
-    if submit_button:
-        # Add user input to chat history
-        st.session_state["chat_history"].append({"role": "user", "content": user_input})
-
-        # Generate Sherlock's response
-        response = model.generate_content([sherlock_persona, sherlock_guidelines, user_input])
-        st.session_state["chat_history"].append({"role": "assistant", "content": response.text})
-
-        # Clear input text field
-        st.session_state["user_input"] = ""
-
-    with output_container:
-        for chat_entry in st.session_state["chat_history"]:
-            if chat_entry["role"] == "user":
-                st.markdown(f"**You:** {chat_entry['content']}")
-            else:
-                st.markdown(f"**Sherlock:** {chat_entry['content']}")
-
 def main():
     # --- Vintage Sherlock Holmes Theme ---
     st.set_page_config(page_title="AI Detective Sherlock Holmes", page_icon=":mag_right:")
@@ -352,7 +319,12 @@ def main():
     if choice == "Investigate Case":
         investigate()
     else:
-        chat_with_sherlock()
+        # Chat with Sherlock Holmes (Gemini 1.5 Pro)
+        st.write("No case files uploaded. Feel free to chat with Sherlock Holmes.")
+        user_query = st.text_input("Ask Sherlock:")
+        if user_query:
+            response = model.generate_content([sherlock_persona, sherlock_guidelines, user_query])
+            st.write(response.text)
 
 if __name__ == "__main__":
     main()