Update sherlock2.py

sherlock2.py

@@ -213,173 +213,118 @@ def search_internet(case_text):
 
     return internet_search_results
 
-def investigate(uploaded_documents, uploaded_images):
-    """ Handles the case investigation process, including text extraction, embedding generation, 
-     image processing, information analysis using Gemini models, web/Wikipedia search, and case report generation.
+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")
+
+    # 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
+
+        # Generate a case report in Sherlock Holmes's style
+        report_prompt = """
+        You are Sherlock Holmes, the renowned detective. Based on the case information, your analysis, findings from 
+        Wikipedia and the web, and the extracted keywords, generate a comprehensive case report in your signature style, 
+        including deductions, potential suspects, and conclusions. 
+        """
+
+        final_report = model.generate_content([sherlock_persona, sherlock_guidelines, report_prompt, 
+                                               *case_embeddings, str(wikipedia_info), str(web_search_results)])
+        st.header("Case Report")
+        st.write(final_report.text)
 
-    # 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])
-    sherlock_analysis = response.text  # Store Sherlock 's analysis
-
-    # 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
-
-    # Generate a case report  in Sherlock Holmes's style
-    report_prompt = """
-    You are Sherlock Holmes, the renowned detective. Based on the case information, your analysis, findings from 
-    Wikipedia and the web, and the extracted keywords, generate a comprehensive case report in your signature style, 
-    including deductions,  potential suspects, and conclusions. 
-    """
-
-    final_report = model.generate_content([sherlock_persona, sherlock_guidelines, report_prompt, 
-                                           *case_embeddings, str(wikipedia_info), str(web_search_results)])
-    
-    return  {
-        "sherlock_analysis": sherlock_analysis,
-        "wikipedia_info": wikipedia_info,
-        "web_search_results": web_search_results,
-        "final_report": final_report.text
-    }
-
-def chat_with_sherlock(user_query, chat_history=None):
-    """Handles the  chat interaction with Sherlock Holmes (Gemini 1.0 Pro)."""
-    if chat_history is None:
-        chat_history = []
-
-    # 1. Create a list of 'Part' objects from the chat history 
-    chat_parts = [
-        glm.Part(text=f"{message['role']}: {message['content']}")
-        for message in chat_history
-    ]
-
-    # 2. Add the user  query as a new Part
-    chat_parts.append(glm.Part(text=f"user: {user_query}"))
-
-    # 3. Generate response using the list of Parts
-    response = model.generate_content(
-        [sherlock_persona, sherlock_guidelines , glm.Content(parts=chat_parts)]
-    )
-
-    # 4. Add Sherlock's response to chat history
-    chat_history.append({"role": "assistant", "content": response.text})
-
-    return response.text, chat_history
-
-
-
-def display_chat_history(): 
-    """Displays the chat history between the user and Sherlock."""
-    for user_msg, bot_msg in st.session_state.chat_history:
-        st.info(f"**You:** {user_msg}")
-        st.success(f"**Sherlock:** {bot_msg}")
-
-
+    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:")
+        if user_query:
+            response = model.generate_content([sherlock_persona, sherlock_guidelines, user_query])
+            st.write(response.text)
 def main():
     # --- Vintage Sherlock Holmes Theme ---
     st.set_page_config(page_title="AI Detective Sherlock Holmes", page_icon=":mag_right:")
 
-    # Custom CSS for Styling (with fixed input at the bottom)
-    vintage_css = """  
+    # Custom CSS for Styling
+    vintage_css = """
     <style>
     body {
         background-color: #d2b48c; /* Antique White */
         color: #332200; /* Dark Brown */
-        font- family: 'Times New Roman', serif; 
+        font-family: 'Times New Roman', serif; 
     }
     h1, h2, h3 {
         color: #8b4513; /* Saddle Brown */
     }
     .stTextInput > div > div > input {
-        border: 1px solid # 8b4513;
-        border-radius: 5px;  
+        border: 1px solid #8b4513;
+        border-radius: 5px; 
     }
-    .stButton > button { 
+    .stButton > button {
         background-color: #8b4513;
         color: white;
         border: none;
-        border-radius: 5 px;
-    }
-    /* Fixed input at the bottom */
-    footer {
-        position: fixed;
-        bottom: 0;
-        width: 100%;
-        background-color: #eee;
-        padding: 10px;
+        border-radius: 5px;
     }
-    </ style>
+    </style>
     """
-    st.markdown(vintage_css, unsafe_allow_html=True)
+    st.markdown(vintage_css, unsafe_allow_html=True)  # Apply custom CSS
 
     # Title and Header
     st.title("AI Detective Sherlock Holmes")
     st.header("_'Elementary, my dear Watson!'_")
 
-    # Sidebar for file uploads
-    st.sidebar.title("Case Files")
-    uploaded_documents = st.sidebar.file_uploader("Upload Documents", accept_multiple_files=True, type=["txt", "pdf", "docx"])
-    uploaded_images = st.sidebar.file_uploader("Upload Images", accept_multiple_files=True, type=["jpg", "png", "jpeg"])
-
-    # Initialize chat history
-    if "chat_history" not in st.session_state:
-        st.session_state.chat_history = []
-
-    # Display chat history above the input box
-    for message  in st.session_state.chat_history:
-        if message["role"] == "user":
-            st.info(f"**You:** {message['content']}")
-        else:
-            st.success(f"**Sherlock:** {message['content']}")
-
-    # Input  box in the footer
-    with st.container():  # Create a container for better visual separation
-        user_query = st.text_input("Ask Sherlock:", key="sherlock_input")
-
-    # Case investigation button
-    if uploaded_documents and uploaded_images and st.button("Analyze  Case"):
-        results = investigate(uploaded_documents, uploaded_images)
-        st.header("Case Report")
-        st.write(results["final_report"])
-
-    # Chat with Sherlock
-    if user_query:
-        response, st.session_state.chat_history =  chat_with_sherlock(user_query, st.session_state.chat_history)
-        st.session_state.chat_history.append({"role": "assistant", "content": response})
-        # Redisplay chat history (to update the UI)
-        for message in st .session_state.chat_history:
-            if message["role"] == "user":
-                st.info(f"**You:** {message['content']}")
-            else:
-                st.success(f"**Sherlock:** {message['content']}") 
-
-
-if __name__  == "__main__":
+    # Add a sidebar for navigation
+    st.sidebar.title("Navigation")
+    options = ["Investigate Case", "Chat with Sherlock"]
+    choice = st.sidebar.radio("Choose an option:", options)
+
+    if choice == "Investigate Case":
+        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:")
+        if user_query:
+            response = model.generate_content([sherlock_persona, sherlock_guidelines, user_query])
+            st.write(response.text)
+
+if __name__ == "__main__":
     main()