Update app.py

app.py

@@ -55,8 +55,32 @@ def google_search(term, num_results=5, lang="en", timeout=5, safe="active", ssl_
         attempts = 0
         while len(all_results) < num_results and attempts < max_attempts:
             try:
-                # ... (Google search request remains the same) ...
+                # Choose a random user agent
+                user_agent = random.choice(_useragent_list)
+                headers = {'User-Agent': user_agent}
+
+                resp = session.get(
+                    url="https://www.google.com/search",
+                    headers=headers,
+                    params={
+                        "q": search_term,
+                        "num": num_results - len(all_results),
+                        "hl": lang,
+                        "start": start,
+                        "safe": safe,
+                    },
+                    timeout=timeout,
+                    verify=ssl_verify,
+                )
+                resp.raise_for_status()
+
+                soup = BeautifulSoup(resp.text, "html.parser")
+                result_block = soup.find_all("div", attrs={"class": "g"})
                 
+                if not result_block:
+                    print("No more results found.")
+                    break
+
                 for result in result_block:
                     if len(all_results) >= num_results:
                         break
@@ -187,7 +211,7 @@ def summarize_financial_news(query):
 Provide a detailed, coherent summary focusing on financial implications and analysis."""
 
         summary = query_llama({"inputs": prompt, "parameters": {"max_length": 1000}})
-        if summary and 'generated_text' in summary[0]:
+        if summary and isinstance(summary, list) and 'generated_text' in summary[0]:
             summaries.append(summary[0]['generated_text'])
     
     # Combine summaries
@@ -200,7 +224,7 @@ Focus on the most important financial implications and analysis."""
 
     final_summary = query_llama({"inputs": final_prompt, "parameters": {"max_length": 2000}})
     
-    if final_summary and 'generated_text' in final_summary[0]:
+    if final_summary and isinstance(final_summary, list) and 'generated_text' in final_summary[0]:
         return final_summary[0]['generated_text']
     else:
         return "Unable to generate summary due to an error."