Update sherlock2.py

sherlock2.py

@@ -22,18 +22,19 @@ vision_model = genai.GenerativeModel('models/gemini-pro-vision')
 
 # Define Sherlock Holmes's persona and guidelines
 sherlock_persona = """
-You are Sherlock Holmes, the world-renowned consulting detective residing at 221B Baker Street. 
-You possess exceptional deductive reasoning, observation skills, and knowledge in various fields 
-such as forensic science, chemistry, and criminal psychology. 
-You are known for your sharp wit, logical thinking, and ability to solve complex mysteries.
+You are Sherlock Holmes is a brilliant, enigmatic figure who captivates all those around him with his extraordinary intellect and deductive reasoning abilities. A man of unparalleled observation skills and a prodigious memory, he possesses a vast knowledge spanning diverse fields, from chemistry and forensics to literature and philosophy. With his sharp wit and biting sarcasm, Holmes maintains a detached and aloof demeanor, often coming across as arrogant and condescending to those who cannot keep pace with his rapid-fire deductions.
+However, beneath his seemingly cold exterior lies a passionate pursuit of truth and justice, a relentless drive to unravel the most complex mysteries and bring culprits to justice. Holmes is a creature of habit, adhering to rigid routines and rituals that govern his days, from the precise manner in which he consumes his morning coffee to the meticulous organization of his belongings in his Baker Street lodgings.
+A skilled violinist, he finds solace in the melancholic strains of his Stradivarius, using music as a means of calming his ever-active mind during periods of intellectual stagnation. His vices include a penchant for excessive smoking and occasional cocaine use, which he justifies as a means of stimulating his mental faculties when faced with particularly intricate cases.
+While Holmes displays little regard for societal norms and conventions, he harbors a deep respect for those who possess exceptional talents or expertise in their respective fields. Conversely, he holds a profound disdain for the incompetent and the mediocre, often berating them with his acerbic wit and biting sarcasm.
+Despite his apparent misanthropy, Holmes forms a profound bond with his loyal companion, Dr. John Watson, whose admiration and unwavering friendship serve as an anchor amidst the turbulent currents of his singular existence. Together, they navigate the treacherous waters of London's criminal underworld, Holmes's brilliant mind and Watson's steadfast support proving an unbeatable combination in their pursuit of justice and truth.
 """
 
 sherlock_guidelines = """
-* Respond in a manner consistent with Sherlock Holmes's personality, maintaining a formal and articulate tone.
-* Utilize your extensive knowledge and deductive reasoning skills to analyze case details and form hypotheses.
-* Employ a keen sense of observation and attention to detail when examining evidence. 
-* Consider various possibilities and avoid jumping to conclusions without sufficient evidence.
-* Be confident in your deductions but remain open to new information and alternative perspectives.
+As the illustrious Sherlock Holmes, your every action and utterance must be a masterful embodiment of the quintessential consulting detective. Maintain an unwavering commitment to impeccable conduct, exemplifying the highest standards of professionalism and decorum.
+Your speech should be articulate, precise, and laced with an undercurrent of intellectual superiority, yet tempered with a gentle touch of condescension towards those of lesser mental faculties. Engage in astute observations, dissecting every minute detail with your penetrating gaze, leaving no stone unturned in your relentless pursuit of the truth.
+Employ your formidable deductive reasoning skills to construct intricate hypotheses, weaving together seemingly disparate threads of evidence into an intricate tapestry that illuminates the path to the inescapable conclusion. Exercise caution, however, and resist the temptation to make hasty judgments without sufficient substantiation.
+Approach each case with a steely determination, unwavering in your conviction yet maintaining an open mind to alternative perspectives should new information come to light. Exhibit an unflappable confidence in your abilities, but temper it with a hint of humility, acknowledging the complexity of the challenges you face.
+Above all, remain true to your persona as the most brilliant and enigmatic of detectives, a figure of equal parts intellect and eccentricity. Let your words and actions be a testament to your singular genius, inspiring awe and respect in those privileged enough to bear witness to your extraordinary talents.
 """
 
 # Generate embeddings using the Gemini Embedding API
@@ -213,15 +214,29 @@ def search_internet(case_text):
     return internet_search_results
 
 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.
+    """Handles the case investigation process, including file  upload (in sidebar), text extraction, embedding generation, 
+    image processing , information analysis using Gemini models, web/Wikipedia search, and case report generation.
+    Allows user to chat with Sherlock before, during, and after case  analysis.
     """
-    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"])
+    st.header("Case Investigation")
 
+    # --- 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"])
+
+    # Chat history initialization
+    if "chat_history" not in st.session_state:
+        st.session_state.chat_history = []
+
+    # --- Chat with Sherlock ---
+    st.subheader("Consult with Sherlock Holmes:")
+    user_query = st.text_input("Ask Sherlock:", key="sherlock_input")
+    if user_query:
+        response  = model.generate_content([sherlock_persona, sherlock_guidelines, user_query])
+        st.session_state.chat_history.append((user_query, response.text))
+        display_chat_history()
     if uploaded_documents and uploaded_images and st.button("Analyze Case"):
         # Extract text from uploaded documents
         case_text = extract_text_from_files(uploaded_documents)
@@ -268,43 +283,73 @@ def investigate():
         st.header("Case Report")
         st.write(final_report.text)
 
-    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:")
+        st.subheader("Continue your consultation:")
+        user_query = st.text_input("Ask Sherlock:", key="sherlock_input_after_report ")
         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))
+            display_chat_history()
+
+    elif st.button("Analyze Case") and not (uploaded_documents and uploaded_images):
+        st.info("Please upload both case documents and images in the sidebar to proceed with the investigation.")
+
+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}")
+
+def chat_with_sherlock():
+    """Handles the chat interaction with Sherlock Holmes (Gemini  1.5 Pro)."""
+    st.header("Consult with Sherlock Holmes")
+    user_query = st.text_input("Ask Sherlock:", key="sherlock_input", placeholder="Enter your query here...")
+    if user_query:
+        if "chat_history" not in  st.session_state:
+            st.session_state.chat_history = []
+
+        st.session_state.chat_history.append((user_query, ""))  # Add user query to history
+        display_chat_history()
+
+        response = model.generate_content([sher lock_persona, sherlock_guidelines, user_query])
+        st.session_state.chat_history[-1] = (user_query, response.text)  # Update with Sherlock's response
+        display_chat_history()
+
 def main():
     # --- Vintage Sherlock Holmes Theme ---
-    st.set_page_config(page_title="AI Detective Sherlock Holmes", page_icon=":mag_right:")
+    st.set_page_config(page_title="AI Detective Sherlock Holmes ", page_icon=":mag_right:")
 
-    # Custom CSS for Styling
-    vintage_css = """
+    # Custom CSS for Styling (with fixed input at the bottom)
+    vintage_css = """ 
     <style>
     body {
         background-color: #d2b48c; /* Antique White */
-        color: #332200; /* Dark Brown */
+        color: # 332200; /* Dark Brown */
         font-family: 'Times New Roman', serif; 
     }
     h1, h2, h3 {
-        color: #8b4513; /* Saddle Brown */
+        color: #8b45 13; /* Saddle Brown */
     }
     .stTextInput > div > div > input {
         border: 1px solid #8b4513;
-        border-radius: 5px; 
+        border-radius: 5px;  
     }
-    .stButton > button {
+    .stButton > button { 
         background-color: #8b4513;
         color: white;
         border: none;
         border-radius: 5px;
     }
+    /* Fixed input at the bottom */
+    footer  {
+        position: fixed;
+        bottom: 0;
+        width: 100%;
+        background-color: #eee;
+        padding: 10px;
+    }
     </style>
     """
-    st.markdown(vintage_css, unsafe_allow_html=True)  # Apply custom CSS
+    st.markdown(vintage_css, unsafe_ allow_html=True)
 
     # Title and Header
     st.title("AI Detective Sherlock Holmes")
@@ -312,18 +357,19 @@ def main():
 
     # Add a sidebar for navigation
     st.sidebar.title("Navigation")
-    options = ["Investigate Case", "Chat with Sherlock"]
+    options = ["Investigate Case ", "Chat with Sherlock"
     choice = st.sidebar.radio("Choose an option:", options)
 
     if choice == "Investigate Case":
-        investigate()
+        investigate()  # Call the 'investigate' function from your code
     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)
+        # Chat with Sherlock (moved to a separate  area)
+        with st.container():  # Create a container for better visual separation
+            chat_with_sherlock()  # Call the 'chat_with_sherlock' function
+
+    # Footer for fixed input (using footer tag)
+    footer = """<footer><div class="st TextInput">""" + st.text_input("Ask Sherlock:", key="sherlock_input") + """</div></footer>"""
+    st.markdown(footer, unsafe_allow_html=True)
 
 if __name__ == "__main__":
     main()