google-local-results-ai-parser 0.2.2 → 0.2.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 78ecf24cc717254a2d3b597868e582f715fc86c7ca160aadbdef5183a3cdb9e7
-  data.tar.gz: '09818139bdc702f86693ff1b48519b6ecbc1d4a5264d9969faf315a04fa901f2'
+  metadata.gz: 62925be13b2ff224585862444f2e77bdea9f7975316b2f66253e7e2fa7a6fe34
+  data.tar.gz: e2ec29030b6c442f955b3d4c54b21473a23f214d3db0944e2006c6ac3c7c9446
 SHA512:
-  metadata.gz: e69e9916971731c5461b3060c558b84e5373a411197b7cd90252601f9ce35d8b07ab9854a5d8663ea6b9f70c21d94f4aa37e182e0956e320e44c687c36ef06e9
-  data.tar.gz: 1e34f84316101ba580189b4b783f42a20c814f3ae1b2e4337a7f46373c842de492bde9f54448e86f02432327309649d0a38d973129ac946bbe3bc39b193f3f4c
+  metadata.gz: 48054b17769730be43150502b909a82666adca5f35b2bf6fa235cf4b96da37040338eb41c62382da0320729158b3d70898efb6d595ad6cef7363d212aa4414f1
+  data.tar.gz: af7d2196995a5991c017940af28de5816e1fc9535b1424c070670db65cf4f6df546d76d2c77ed0f64281810dda649f769e7ead071e5c99c68f43a18c4dbf89ee

data/lib/google-local-results-ai-parser.rb

@@ -2,6 +2,7 @@ require 'nokolexbor'
 require 'http'
 require 'parallel'
 require 'json'
+require 'pry'
 
 module GoogleLocalResultsAiParser
   DEFAULT_SERVER = 'https://api-inference.huggingface.co/models/serpapi/bert-base-local-results'.freeze
@@ -465,16 +466,17 @@ module GoogleLocalResultsAiParser
         "Takeaway"
       ]
       caught_results_indices = results.map.with_index {|result, index| index if known_errors.include?(result[:input])}.compact
-      return results, label_order, duplicates if caught_results_indices == []
 
-      not_service_option_duplicates = duplicates.select.with_index do |duplicate, duplicate_index|
-        caught_results_indices.each do |caught_index|
+      not_service_option_duplicates = []
+      caught_results_indices.each do |caught_index|
+        duplicates.each.with_index do |duplicate, duplicate_index|
           if duplicate.include?(caught_index) && results[caught_index][:result][0][0]["label"] != "service_options"
-            duplicate_index
+            not_service_option_duplicates << duplicate_index
           end
         end
       end
 
+      return results, label_order, duplicates if not_service_option_duplicates == []
       # Zero out the `type` or `description`, and put it to last position
       caught_results_indices.each do |caught_index|
         service_options_hash = results[caught_index][:result][0].find {|hash| hash["label"] == "service options" }
@@ -491,12 +493,12 @@ module GoogleLocalResultsAiParser
 
       # Rearranging duplicates
       not_service_option_duplicates.each do |not_service_option_duplicate|
-        last_item = duplicates[duplicates.index(not_service_option_duplicate)][-1]
-        duplicates[duplicates.index(not_service_option_duplicate)].delete(last_item)
+        last_item = duplicates[not_service_option_duplicate][-1]
+        duplicates[not_service_option_duplicate].delete(last_item)
       end
 
       not_service_option_duplicates.each do |not_service_option_duplicate|
-        if (duplicate_arr = duplicates[duplicates.index(not_service_option_duplicate)]) && duplicate_arr.size == 1
+        if (duplicate_arr = duplicates[not_service_option_duplicate]) && duplicate_arr.size == 1
           duplicates.delete(duplicate_arr)
         end
       end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: google-local-results-ai-parser
 version: !ruby/object:Gem::Version
-  version: 0.2.2
+  version: 0.2.3
 platform: ruby
 authors:
 - Emirhan Akdeniz