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

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: c96be517ff3f5cc269d4fb60fe3db8309393f814d084070827fc056f00ccd21e
-  data.tar.gz: fdb3da5e1c1561373dc3898441c3f85e410ce54b50867aa560728ce3366710a2
+  metadata.gz: 7b078ffd7df8466aa1859da5120725c18b9425ddb331a16a54c913811af0d3b8
+  data.tar.gz: 637c78ed81d02642834e083b1bbcbab1962a09afd218e490e02ffc9ae0232013
 SHA512:
-  metadata.gz: 30206bdb81631006ff75f44ccc8917ecf3f0b87b85f7efdefe4b0d6474ee761eb49466e0edd1c1b9a2a552f9fa4bac0df61d42ef7126fa2196c3761adc0fa973
-  data.tar.gz: 228118960a094c140a7a7d8a2a63a7be6db481b08170ae16346b4107cce4f82d5a1a9c7edc1d28f1d479ad21975ccaae4768faebac2541d50a6ed114c75de280
+  metadata.gz: 216e1a13a42b32294a5196f26bd34a05f05cf53cfbf4b50f05b2f7ba93f1f8349a1eff7becd6900f6d96e2c18af910af96be8c92b6a5e467262d672aff4e97d5
+  data.tar.gz: 3dc359ac52dd5e3475bdb4817135f17cc98621a1a59d21078c13090d3e16d6bf10b37520c627fbc0e203f3c6fa51257bb3ae682e65e8b146b5eb8bc1ed187a13

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

@@ -94,7 +94,10 @@ module GoogleLocalResultsAiParser
         results, label_order, duplicates = description_as_hours_confusion(results, label_order, duplicates)
         results, label_order, duplicates = description_as_type_confusion(results, label_order, duplicates)
         results, label_order, duplicates = reviews_as_rating_confusion(results, label_order, duplicates)
+        results, label_order, duplicates = reviews_as_price_confusion(results, label_order, duplicates)
         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)
         
         # General clashes
         line_result = check_if_on_different_lines(results, duplicates, unsplit_text)
@@ -240,6 +243,73 @@ module GoogleLocalResultsAiParser
       return results, label_order, duplicates
     end
 
+    # 104 Ave ... Share
+    # Fixes `Share`
+    def button_text_as_address_confusion(results, label_order, duplicates)
+      known_errors = ["Share"]
+      confusion_condition = results.any? {|result| known_errors.include?(result[:input])}
+      return results, label_order, duplicates unless confusion_condition
+
+      address_duplicate = duplicates.find.with_index do |duplicate, duplicate_index|
+        if results[duplicate[0]][:result][0][0]["label"] == "address"
+          duplicate_index
+        end
+      end
+
+      # Delete the known button text directly
+      results.delete_at(address_duplicate[-1])
+      
+      # Rearranging `label_order`
+      label_order.delete_at(address_duplicate[-1])
+      
+      # Rearranging duplicates
+      last_item = duplicates[duplicates.index(address_duplicate)][-1]
+      duplicates[duplicates.index(address_duplicate)].delete(last_item)
+
+      if (duplicate_arr = duplicates[duplicates.index(address_duplicate)]) && duplicate_arr.size == 1
+        duplicates.delete(duplicate_arr)
+      end
+
+      return results, label_order, duplicates
+    end
+
+    # Order pickup
+    # Fixes `Order pickup`
+    def button_text_as_service_options_confusion(results, label_order, duplicates)
+      known_errors = ["Order pickup"]
+      confusion_condition = results.any? {|result| known_errors.include?(result[:input])}
+      return results, label_order, duplicates unless confusion_condition
+
+      service_options_indexes = results.map {|result| results.index(result) if known_errors.include?(result[:input])}.compact
+
+      service_options_duplicate = duplicates.find.with_index do |duplicate, duplicate_index|
+        if results[duplicate[0]][:result][0][0]["label"] == "service options"
+          duplicate_index
+        end
+      end
+
+      # Delete the known button text directly
+      service_options_indexes.each {|index| results.delete_at(index)}
+      
+      # Rearranging `label_order`
+      service_options_indexes.each {|index| label_order.delete_at(index)}
+      
+      # Rearranging duplicates
+      service_options_indexes.each do |index|
+        duplicates.each_with_index do |duplicate, duplicate_index|
+          if duplicate.include?(index)
+            duplicates[duplicate_index].delete(index)
+          end
+        end
+      end
+
+      if service_options_duplicate && (duplicate_arr = duplicates[duplicates.index(service_options_duplicate)]) && duplicate_arr.size == 1
+        duplicates.delete(duplicate_arr)
+      end
+
+      return results, label_order, duplicates
+    end
+
     # 3.4 .. (1.4K)
     # Fixes `(1.4K)`
     def reviews_as_rating_confusion(results, label_order, duplicates)
@@ -272,6 +342,38 @@ module GoogleLocalResultsAiParser
       return results, label_order, duplicates
     end
 
+    # (1.6K) .. $
+    # Fixes `(1.6K)`
+    def reviews_as_price_confusion(results, label_order, duplicates)
+      price_duplicate = duplicates.find.with_index do |duplicate, duplicate_index|
+        if results[duplicate[0]][:result][0][0]["label"] == "price"
+          duplicate_index
+        end
+      end
+
+      if price_duplicate && results[price_duplicate[0]][:input][/\(\d+\.\d+\w\)/]
+        # 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}
+        
+        # Rearranging `label_order`
+        label_order[price_duplicate[-1]] = "reviews"
+        
+        # Rearranging duplicates
+        last_item = duplicates[duplicates.index(price_duplicate)][-1]
+        duplicates[duplicates.index(price_duplicate)].delete(last_item)
+
+        if (duplicate_arr = duplicates[duplicates.index(price_duplicate)]) && duplicate_arr.size == 1
+          duplicates.delete(duplicate_arr)
+        end
+      end
+
+      return results, label_order, duplicates
+    end
+
     # Coffee shop ... Iconic Seattle-based coffeehouse chain
     # Fixes `Iconic Seattle-based coffeehouse chain`
     def description_as_type_confusion(results, label_order, duplicates)
@@ -305,7 +407,7 @@ module GoogleLocalResultsAiParser
     # Fixes `Closes 12 AM``
     def description_as_hours_confusion(results, label_order, duplicates)
       description_index = label_order.index("description")
-      hours_index =  label_order.index("hours")
+      hours_index =  label_order.rindex("hours")
 
       # Description may or may not be a duplicate.
       # This is a known error from the model, so it has to be handled in any case.

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: google-local-results-ai-parser
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.1.3
 platform: ruby
 authors:
 - Emirhan Akdeniz
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2023-06-15 00:00:00.000000000 Z
+date: 2023-06-20 00:00:00.000000000 Z
 dependencies: []
 description: A gem to be used with serpapi/bert-base-local-results model to predict
   different parts of Google Local Listings. This gem uses BERT model at https://huggingface.co/serpapi/bert-base-local-results