Update sherlock2.py

sherlock2.py

@@ -164,6 +164,7 @@ def search_and_scrape_wikipedia(keywords, max_topics_per_query=3, react_model='g
 
     search_history = set()
     wikipedia_info = []
+
     for query in keywords:
         # Use ReAct to search and extract information
         react_agent(query) 
@@ -174,20 +175,21 @@ def search_and_scrape_wikipedia(keywords, max_topics_per_query=3, react_model='g
         for line in response_text.strip().split('\n'):
             if line.startswith("Observation"):
                 observations.append(line.split(':')[-1].strip())
-        
-        # Assuming the last observation is the final answer/summary
-        summary = observations[-1] 
-
-        # Get URL from search history (assuming successful search)
-        url = react_agent._search_urls[-1]
-
-        # Create a dictionary with the extracted information
-        wikipedia_info.append({
-            "topic": query,  # Using the original query as the topic
-            "summary": summary,
-            "url": url
-            # "additional_sources":  # Add this if you implement additional source extraction
-        })
+
+        # Check if observations list is not empty before accessing elements
+        if observations:
+            summary = observations[-1]  
+            url = react_agent._search_urls[-1] 
+
+            # Create a dictionary with the extracted information
+            wikipedia_info.append({
+                "topic": query,  # Using the original query as the topic
+                "summary": summary,
+                "url": url
+            })
+        else:
+            # Handle the case where no observations were found (optional)
+            print(f"No observations found for query: {query}")
 
     return wikipedia_info