Update sherlock2.py

sherlock2.py

@@ -249,8 +249,10 @@ def investigate():
         st.header("Case Report")
         st.write(final_report.text)
 
-        # Display chat history and allow user input
+        # Display chat history
         display_chat_history()
+
+        # Render user input box
         user_query = st.text_input("Ask Sherlock:", key="user_input")
         if user_query:
             response = model.generate_content([sherlock_persona, sherlock_guidelines, user_query])
@@ -263,6 +265,8 @@ def investigate():
     else:
         st.info("Please upload both case documents and images to proceed with the investigation.")
         display_chat_history()
+
+        # Render user input box
         user_query = st.text_input("Ask Sherlock:", key="user_input")
         if user_query:
             response = model.generate_content([sherlock_persona, sherlock_guidelines, user_query])