Wajahat698 commited on
Commit
0cacd4b
·
verified ·
1 Parent(s): b7c1be0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +27 -4
app.py CHANGED
@@ -1028,14 +1028,37 @@ def search_knowledge_base(query):
1028
 
1029
  def google_search(query):
1030
  try:
 
1031
  search_client = serpapi.Client(api_key=serper_api_key)
1032
- results = search_client.search({"engine": "google", "q": query})
1033
- snippets = [result["snippet"] for result in results.get("organic_results", [])]
1034
- return snippets
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1035
  except Exception as e:
1036
  logger.error(f"Error in Google search: {e}")
1037
  return ["Error occurred during Google search"]
1038
-
1039
 
1040
 
1041
  # RAG response function
 
1028
 
1029
  def google_search(query):
1030
  try:
1031
+ # Initialize SerpAPI client and set parameters
1032
  search_client = serpapi.Client(api_key=serper_api_key)
1033
+ search_params = {
1034
+ "engine": "google", # Use Google as the search engine
1035
+ "q": query, # Query
1036
+ "sort": "date", # Sort by latest date
1037
+ "hl": "en", # Language set to English
1038
+ "gl": "us" # Country/region set to US
1039
+ }
1040
+
1041
+ # Perform the search
1042
+ results = search_client.search(search_params)
1043
+
1044
+ # Process the results to extract links and snippets
1045
+ organic_results = results.get("organic_results", [])
1046
+ valid_results = []
1047
+ for result in organic_results:
1048
+ link = result.get("link", "")
1049
+ snippet = result.get("snippet", "")
1050
+ # Validate the link (basic check to avoid blank or invalid links)
1051
+ if link.startswith("http"):
1052
+ valid_results.append({"link": link, "snippet": snippet})
1053
+
1054
+ # Ensure we return meaningful results
1055
+ if valid_results:
1056
+ return valid_results
1057
+ return ["No valid links found"]
1058
+
1059
  except Exception as e:
1060
  logger.error(f"Error in Google search: {e}")
1061
  return ["Error occurred during Google search"]
 
1062
 
1063
 
1064
  # RAG response function