Update sherlock2.py

sherlock2.py

@@ -355,38 +355,26 @@ def investigate():
         with st.expander("Sherlock's Analysis and Suggestions:"):
             st.write(response.text)
 
-        # Initialize wikipedia_info with an empty list
-        wikipedia_info = []  # Add this line here
-
-        search_options = st.multiselect("Search for additional clues:", ["Wikipedia", "Internet"], default=["Wikipedia"])
+        search_options = st.multiselect("Search for additional clues:", ["Internet"], default=["Internet"])  # Only "Internet" option
         if st.button("Search"):
             with st.spinner("Searching for clues..."):
-                web_search_results = []
-                if "Wikipedia" in search_options:
-                    wikipedia_info = search_and_scrape_wikipedia(keywords) 
-                    st.subheader("Wikipedia Findings:")
-                    for info in wikipedia_info:
-                        st.write(f"**Topic:** {info['topic']}")
-                        st.write(f"**Summary:** {info['summary']}")
-                        st.write(f"**URL:** {info['url']}")
-                if "Internet" in search_options:
-                    web_search_results = search_internet("\n\n".join(case_text))
-                    st.subheader("Internet Search Results:")
-                    for result in web_search_results:
-                        st.write(f"**Title:** {result['title']}")
-                        st.write(f"**Snippet:** {result['snippet']}")
-                        st.write(f"**URL:** {result['url']}")
+                web_search_results = search_internet("\n\n".join(case_text))
+                st.subheader("Internet Search Results:")
+                for result in web_search_results:
+                    st.write(f"**Title:** {result['title']}")
+                    st.write(f"**Snippet:** {result['snippet']}")
+                    st.write(f"**URL:** {result['url']}")
 
         # Generate report button
         if st.button("Generate Case Report"):
             with st.spinner("Generating report..."):
                 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, 
+                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, str(wikipedia_info), str(web_search_results)])
+                                                       case_embeddings_str, str(web_search_results)])  # Removed wikipedia_info
                 st.header("Case Report")
                 st.write(final_report.text)