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

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 32115a821e2e34cb64ab9e0b28e7c5be3a5b0de03006617c429e094bddd3e38d
-  data.tar.gz: c298704a904a2867528bf4ff9827bf50fdbde288fcffd5d8076447d1817e400e
+  metadata.gz: 62925be13b2ff224585862444f2e77bdea9f7975316b2f66253e7e2fa7a6fe34
+  data.tar.gz: e2ec29030b6c442f955b3d4c54b21473a23f214d3db0944e2006c6ac3c7c9446
 SHA512:
-  metadata.gz: '0782715755059bf87928160366f3bb6168f1b12fab37f9e66cee5cf46a22406d2a05d6fd19e7e17f3f0f0c8925a5cfcdead320f009ed4b58b7f7aff8829ef896'
-  data.tar.gz: c0b8d099c542dcc5e64219594db5f2fa3f9f634835e0f6c2b18818b067d341bee8666045afc9f4c2da889ee86637949d7572f3bd03493395f47d1f5e2de1d023
+  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
@@ -63,7 +64,7 @@ module GoogleLocalResultsAiParser
 
         if final_results[label]
           # Combine the text for same elements
-          final_results[label] = unsplit_text[/#{final_results[label]}[.|\n]+#{value}/]
+          final_results[label] = unsplit_text[/#{final_results[label]}(.+|\n)#{value}/]
         else
           # Directly assign values
           final_results[label] = value
@@ -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.1
+  version: 0.2.3
 platform: ruby
 authors:
 - Emirhan Akdeniz