Update sherlock2.py

sherlock2.py

@@ -110,22 +110,6 @@ def process_images(uploaded_images):
             st.error(f"Error processing image: {e}")
     return image_insights
 
-class WikipediaSearcher:
-    def __init__(self):
-        self._search_history = []
-        self._search_urls = []
-
-    def search(self, query):
-        """Performs a Wikipedia search and returns relevant information."""
-        try:
-            summary = wikipedia.summary(query, sentences=3, auto_suggest=False)
-            wiki_url = wikipedia.page(query, auto_suggest=False).url
-            self._search_history.append(query)
-            self._search_urls.append(wiki_url)
-            return f"**Summary:** {summary}\n**URL:** {wiki_url}"
-        except (DisambiguationError, PageError) as e:
-            return f"Could not find information on '{query}' in Wikipedia."
-
 def search_internet(case_text):
     """Generates search queries using Gemini 1.5 Pro and performs internet searches for case-related information."""
     prompt = """
@@ -178,7 +162,7 @@ def clear_chat():
     st.session_state.chat_history = []
 
 def investigate():
-    """Handles the case investigation process with Pandas for embeddings and automatic Wikipedia search."""
+    """Handles the case investigation process with Pandas for embeddings."""
     st.subheader("Case Investigation")
 
     # File upload with clear labels and progress bars
@@ -206,20 +190,6 @@ def investigate():
             "keywords": keywords
         }
 
-        # --- Wikipedia Search using Keywords ---
-        wikipedia_searcher = WikipediaSearcher()
-        wikipedia_results = []
-        with st.spinner("Searching Wikipedia using keywords..."):
-            for keyword in keywords:
-                result = wikipedia_searcher.search(keyword)
-                wikipedia_results.append(result)
-
-        # Display Wikipedia search results
-        st.subheader("Wikipedia Search Results (Based on Keywords):")
-        for result in wikipedia_results:
-            st.write(result)
-
-        # --- Prompt Generation and Sherlock's Analysis ---
         prompt = """
         You are Sherlock Holmes, the renowned detective. Analyze the following case information and provide insights or 
         suggestions for further investigation:
@@ -231,7 +201,6 @@ def investigate():
         with st.expander("Sherlock's Analysis and Suggestions:"):
             st.write(response.text)
 
-        # --- Web Search ---
         web_search_results = [] 
 
         search_options = st.multiselect("Search for additional clues:", ["Internet"], default=["Internet"]) 
@@ -244,21 +213,16 @@ def investigate():
                     st.write(f"**Snippet:** {result['snippet']}")
                     st.write(f"**URL:** {result['url']}")
 
-        # --- Generate Case Report with Wikipedia and Web Results ---
+        # 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 
-                the web and Wikipedia, 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, 
-                    str(web_search_results), 
-                    str(wikipedia_results)
-                ])
+                """ 
+                final_report = model.generate_content([sherlock_persona, sherlock_guidelines, report_prompt, 
+                                                       str(web_search_results)]) 
                 st.header("Case Report")
                 st.write(final_report.text)