Update sherlock2.py

sherlock2.py

@@ -232,45 +232,14 @@ 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.
     """
-    st.header("Case Investigation")
+    st.subheader("Case Investigation")
 
     # File upload for documents and images
     uploaded_documents = st.file_uploader("Upload Case Documents", accept_multiple_files=True, type=["txt", "pdf", "docx"])
     uploaded_images = st.file_uploader("Upload Case Images", accept_multiple_files=True, type=["jpg", "png", "jpeg"])
 
     if uploaded_documents and uploaded_images and st.button("Analyze Case"):
-        # Extract text from uploaded documents
-        case_text = extract_text_from_files(uploaded_documents)
-        
-        # Extract keywords and important information from the text
-        keywords = extract_keywords_simple("\n\n".join(case_text))
-
-        # Generate embeddings for the extracted text
-        case_embeddings = generate_embeddings_from_documents(case_text)
-
-        # Process images using Gemini Pro Vision
-        image_insights = process_images(uploaded_images)
-
-        # Combine text, image, and keyword information
-        combined_information = {
-            "case_text": case_text,
-            "case_embeddings": case_embeddings,
-            "image_insights": image_insights,
-            "keywords": keywords
-        }
-
-        # Analyze combined information using Gemini 1.5 Pro
-        prompt = """
-        You are Sherlock Holmes, the renowned detective. Analyze the following case information and provide insights or 
-        suggestions for further investigation:
-        """ + str(combined_information)
-
-        response = model.generate_content([sherlock_persona, sherlock_guidelines, prompt, *case_embeddings])
-        st.write(response.text)
-
-        # Search Wikipedia and the web for related information
-        wikipedia_info = search_and_scrape_wikipedia(keywords)
-        web_search_results = search_internet("\n\n".join(case_text))  # Search the web
+        # ... (keep the existing code for case investigation)
 
         # Generate a case report in Sherlock Holmes's style
         report_prompt = """
@@ -284,7 +253,9 @@ def investigate():
         st.header("Case Report")
         st.write(final_report.text)
 
-        user_query = st.text_input("Ask Sherlock:")
+        # Display chat history and allow user input
+        display_chat_history()
+        user_query = st.text_input("Ask Sherlock:", key="user_input")
         if user_query:
             response = model.generate_content([sherlock_persona, sherlock_guidelines, user_query])
             st.write(response.text)
@@ -297,9 +268,9 @@ def investigate():
     else:
         st.info("Please upload both case documents and images to proceed with the investigation.")
 
-        # Chat with Sherlock Holmes (Gemini 1.5 Pro)
-        st.write("Alternatively, you may engage in a conversation with Sherlock Holmes.")
-        user_query = st.text_input("Ask Sherlock:")
+        # Display chat history and allow user input
+        display_chat_history()
+        user_query = st.text_input("Ask Sherlock:", key="user_input")
         if user_query:
             response = model.generate_content([sherlock_persona, sherlock_guidelines, user_query])
             st.write(response.text)
@@ -342,30 +313,20 @@ 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 fixInputToTop() {
-                const inputContainer = document.querySelector('.stTextInput');
-                if (inputContainer) {
-                    inputContainer.style.position = 'sticky';
-                    inputContainer.style.top = '20px';
-                    inputContainer.style.left = '20px';
-                    inputContainer.style.right = '20px';
-                    inputContainer.style.zIndex = '999';
-            }
-        }
-        window.addEventListener('load', fixInputToTop);
-        </script>
-        """,
-        unsafe_allow_html=True
-    )
-
     # Title and Header
     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}")
+
     # Display chat history
     display_chat_history()
 
@@ -378,16 +339,17 @@ def main():
         investigate()
     else:
         # 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:")
+        st.subheader("Chat with Sherlock Holmes")
+        user_query = st.text_input("Ask Sherlock:", key="user_input")
         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)
+    # Call the fixInputToBottom function after rendering the input text box
+    st.markdown("<script>fixInputToBottom();</script>", unsafe_allow_html=True)
+
 
 if __name__ == "__main__":
     main()