google-local-results-ai-parser 0.1.3 → 0.1.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 7b078ffd7df8466aa1859da5120725c18b9425ddb331a16a54c913811af0d3b8
-  data.tar.gz: 637c78ed81d02642834e083b1bbcbab1962a09afd218e490e02ffc9ae0232013
+  metadata.gz: f879eea0293f78de7556e0b7b2c899d7c1460ab75955b3e014c9ec8341c183bd
+  data.tar.gz: 5e4b4ba286d337ac577fa2750344d910ec297b981c0b57c53b25cf94be6fb10f
 SHA512:
-  metadata.gz: 216e1a13a42b32294a5196f26bd34a05f05cf53cfbf4b50f05b2f7ba93f1f8349a1eff7becd6900f6d96e2c18af910af96be8c92b6a5e467262d672aff4e97d5
-  data.tar.gz: 3dc359ac52dd5e3475bdb4817135f17cc98621a1a59d21078c13090d3e16d6bf10b37520c627fbc0e203f3c6fa51257bb3ae682e65e8b146b5eb8bc1ed187a13
+  metadata.gz: 3e6b363af552c4bf34da4ac387830720c98bfbc5c1a2331c2fe1fed8c373d3d0f66bb1bd303b8565d2fcaadeeabfca6969fb4f5a13f25c9efc24f2d44923a288
+  data.tar.gz: 801b68771772460632e704cca817d01a2662b8958415510cce48c57af2b37592493e9826d226b4f5bfd67ef5cb432ba8e7fa9b7bac63b84914791ef4a216f1c9

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

@@ -98,6 +98,7 @@ module GoogleLocalResultsAiParser
         results, label_order, duplicates = button_text_as_hours_confusion(results, label_order, duplicates)
         results, label_order, duplicates = button_text_as_address_confusion(results, label_order, duplicates)
         results, label_order, duplicates = button_text_as_service_options_confusion(results, label_order, duplicates)
+        results, label_order, duplicates = service_options_as_description_or_type_confusion(results, label_order, duplicates)
         
         # General clashes
         line_result = check_if_on_different_lines(results, duplicates, unsplit_text)
@@ -355,16 +356,16 @@ module GoogleLocalResultsAiParser
         # Zero out the `price`, and put it to last position
         reviews_hash = results[price_duplicate[-1]][:result][0].find {|hash| hash["label"] == "reviews" }
         reviews_index = results[price_duplicate[-1]][:result][0].index(reviews_hash)
-        results[price_duplicate[-1]][:result][0][0] = {"label" => "reviews", "score" => 1.0}
-        results[price_duplicate[-1]][:result][0].delete_at(reviews_index)
-        results[price_duplicate[-1]][:result][0] << {"label" => "price", "score" => 0.0}
+        results[price_duplicate[0]][:result][0][0] = {"label" => "reviews", "score" => 1.0}
+        results[price_duplicate[0]][:result][0].delete_at(reviews_index)
+        results[price_duplicate[0]][:result][0] << {"label" => "price", "score" => 0.0}
         
         # Rearranging `label_order`
-        label_order[price_duplicate[-1]] = "reviews"
+        label_order[price_duplicate[0]] = "reviews"
         
         # Rearranging duplicates
-        last_item = duplicates[duplicates.index(price_duplicate)][-1]
-        duplicates[duplicates.index(price_duplicate)].delete(last_item)
+        first_item = duplicates[duplicates.index(price_duplicate)][0]
+        duplicates[duplicates.index(price_duplicate)].delete(first_item)
 
         if (duplicate_arr = duplicates[duplicates.index(price_duplicate)]) && duplicate_arr.size == 1
           duplicates.delete(duplicate_arr)
@@ -437,6 +438,49 @@ module GoogleLocalResultsAiParser
       return results, label_order, duplicates
     end
 
+    # On-site services, Online appointments
+    # Fixes `On-site services`, `Online appointments`
+    def service_options_as_description_or_type_confusion(results, label_order, duplicates)
+      known_errors = ["On-site services", "On-site services not available", "Online appointments", "Online appointments not available"]
+      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_duplicate = duplicates.find.with_index do |duplicate, duplicate_index|
+        caught_results_indices.each do |caught_index|
+          if results[caught_index][:result][0][0]["label"] != "service_options"
+            duplicate_index
+          end
+        end
+      end
+
+      # 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" }
+        service_options_index = results[caught_index][:result][0].index(service_options_hash)
+        old_result_hash = results[caught_index][:result][0][0]
+        results[caught_index][:result][0][0] = {"label" => "service options", "score" => 1.0}
+        results[caught_index][:result][0].delete_at(service_options_index)
+        old_result_hash["score"] = 0.0
+        results[caught_index][:result][0] << old_result_hash
+      end
+
+      # Rearranging `label_order`
+      caught_results_indices.each {|caught_index| label_order[caught_index] = "service_options"}
+
+      # Rearranging duplicates
+      not_service_option_duplicate.each do |duplicate_index|
+        duplicate_arr = duplicates.find{|duplicate| duplicate.include?(2)}
+        last_item = duplicate_arr[-1]
+        duplicates[duplicates.index(duplicate_arr)].delete(last_item)
+      end
+
+      if (duplicate_arr = duplicates[duplicates.index(not_service_option_duplicate)]) && duplicate_arr.size == 1
+        duplicates.delete(duplicate_arr)
+      end
+
+      return results, label_order, duplicates
+    end
+
     # Takeaway ⋅ Dine-in ...
     # Fixes `Takeaway`
     def service_options_as_type_confusion(results, label_order, duplicates)

metadata

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