in demo

app.py

@@ -12,110 +12,110 @@ import gradio as gr
 from ultralytics import YOLO
 
 
-search_google = False
-# GOOGLE_API_KEY = "AIzaSyCD5eIu3N2uD6Vf5GE0VLBpkPsJ3mEZgA0"
-# GOOGLE_CSE_ID = "b536303d151b94476"
-# GOOGLE_API_KEY = "AIzaSyBUvPKAZa2TddKYDJI9t0ac2A_JJqdgWHo"
-# GOOGLE_CSE_ID = "f3446c60c52cd4dc2"
-# GOOGLE_API_KEY = "AIzaSyCoBYZK4khX_W5eW-AqEXKOclUIer6CiC8"
-# GOOGLE_CSE_ID = "c3ba01991c1a14c7d"
-mode = 'local'
-# mode = 'rbf'
-
-if mode == 'local':
-    model = YOLO('/Users/selimgilon/Desktop/SportBoost/Projects/Iflyers/best-flyers-yolov8-1000img-v10.pt')
-else:
-    rf = Roboflow(api_key="TN586SQsdwjUUvMN3H8e")
-    project = rf.workspace().project("smartflyers")
-    model = project.version(9).model
-
-# def process(selected_shop):
-#     print(selected_shop)
-#     return selected_shop
-hotspots_sku_links = []
-shop = None
-shops = {
-    'Latulippe': {
-        'url': 'https://latulippe.com/fr/recherche/?keywords=',
-        'regex': r'N° (M\d+)',
-        'multi_items': False,
-    },
-    'Decathlon': {
-        'url': 'https://www.decathlon.fr/search?Ntt=',
-        'regex': r'Réf\. ?: ?([\d, ]+)',
-        'multi_items': True,
-        'brands': [
-            "Allsix", "Aptonia", "Artengo", "Atorka", "B'Twin", "Caperlan", "Domyos", "Dreamscape",
-            "FLX", "Forclaz", "Fouganza", "Geologic", "Geonaute", "Inesis", "Imviso", "Inovik",
-            "Itiwit", "KOODZA", "KOROK", "Kalenji", "Kimjaly", "Kipsta", "Kuikma", "Nabaiji",
-            "Newfeel", "Offload", "Olaian", "Opfeel", "Orao", "Oroks", "Outshock", "Oxelo",
-            "Perfly", "Pongori", "Quechua", "Rockrider", "Sandever", "Simond", "Solognac",
-            "Subea", "Tarmak", "Triban", "Tribord", "Urball", "Van Rysel", "Wed'Ze", "Decathlon"
-        ]
-    },
-    'Canadian Tire': {
-        'url': 'https://www.canadiantire.ca/en/search-results.html?q=',
-        'regex': r'(\d{2,3}-\d{4}-\d(?:/\d{4}-\d)?)',
-        'multi_items': True,
-    },
-    'Walmart': {
-        # 'url': 'https://www.walmart.ca/en/search?q=',
-        'url': '',
-        'regex': r'N°(\d+)',
-        'multi_items': False,
-    },
-    # Add more shops and their respective information here
-}
-
-regex_price = r'(\d+\.?\d*) ?([€$])|([€$]) ?(\d+\.?\d*)'
-
-
-# def app(img, selected_shop):
-#     if img is None:
-#         raise gr.Error("No file uploaded. Please upload an image.")
-#
-#     # Check if the uploaded file is a PDF
-#     if selected_shop is None:
-#         raise gr.Error("Please select a shop.")
-#
-#     try:
-#         image_annotated = section(img)
-#         pass
-#     except Exception as e:
-#         return gr.Error(f"An error occurred: {str(e)}")
-#
-#     return image_annotated, hotspots_sku_links[selected_shop]
-
-
-def build_request(query, start=1, num=10, date_restrict=None, **params):
-    request = {
-        'key': GOOGLE_API_KEY,
-        'q': query,
-        'cx': GOOGLE_CSE_ID,
-        # 'start': start,
-        # 'num': num,
-        # 'dateRestrict': date_restrict,
-    }
-    request.update(params)
-    return request
-
 
-def make_request(request):
-    response = requests.get('https://www.googleapis.com/customsearch/v1', params=request)
-    # print(f"Status Code: {response.status_code}")  # Print status code for debugging
-    if response.status_code != 200:
-        raise Exception("Google Custom Search request failed with status code {}".format(response.status_code))
 
-    search_results = response.json()
-    # print(f"Response JSON: {search_results}")  # Print raw response for debugging
-    if 'items' in search_results and len(search_results['items']) > 0:
-        first_result_url = search_results['items'][0]['link']
-        return first_result_url
+with gr.Blocks() as demo:
+    search_google = False
+    GOOGLE_API_KEY = "AIzaSyCD5eIu3N2uD6Vf5GE0VLBpkPsJ3mEZgA0"
+    GOOGLE_CSE_ID = "b536303d151b94476"
+    # GOOGLE_API_KEY = "AIzaSyBUvPKAZa2TddKYDJI9t0ac2A_JJqdgWHo"
+    # GOOGLE_CSE_ID = "f3446c60c52cd4dc2"
+    # GOOGLE_API_KEY = "AIzaSyCoBYZK4khX_W5eW-AqEXKOclUIer6CiC8"
+    # GOOGLE_CSE_ID = "c3ba01991c1a14c7d"
+    mode = 'local'
+    # mode = 'rbf'
+
+    if mode == 'local':
+        model = YOLO('/Users/selimgilon/Desktop/SportBoost/Projects/Iflyers/best-flyers-yolov8-1000img-v10.pt')
     else:
-        return None
+        rf = Roboflow(api_key="TN586SQsdwjUUvMN3H8e")
+        project = rf.workspace().project("smartflyers")
+        model = project.version(9).model
+
+    # def process(selected_shop):
+    #     print(selected_shop)
+    #     return selected_shop
+    hotspots_sku_links = []
+    shop = None
+    shops = {
+        'Latulippe': {
+            'url': 'https://latulippe.com/fr/recherche/?keywords=',
+            'regex': r'N° (M\d+)',
+            'multi_items': False,
+        },
+        'Decathlon': {
+            'url': 'https://www.decathlon.fr/search?Ntt=',
+            'regex': r'Réf\. ?: ?([\d, ]+)',
+            'multi_items': True,
+            'brands': [
+                "Allsix", "Aptonia", "Artengo", "Atorka", "B'Twin", "Caperlan", "Domyos", "Dreamscape",
+                "FLX", "Forclaz", "Fouganza", "Geologic", "Geonaute", "Inesis", "Imviso", "Inovik",
+                "Itiwit", "KOODZA", "KOROK", "Kalenji", "Kimjaly", "Kipsta", "Kuikma", "Nabaiji",
+                "Newfeel", "Offload", "Olaian", "Opfeel", "Orao", "Oroks", "Outshock", "Oxelo",
+                "Perfly", "Pongori", "Quechua", "Rockrider", "Sandever", "Simond", "Solognac",
+                "Subea", "Tarmak", "Triban", "Tribord", "Urball", "Van Rysel", "Wed'Ze", "Decathlon"
+            ]
+        },
+        'Canadian Tire': {
+            'url': 'https://www.canadiantire.ca/en/search-results.html?q=',
+            'regex': r'(\d{2,3}-\d{4}-\d(?:/\d{4}-\d)?)',
+            'multi_items': True,
+        },
+        'Walmart': {
+            # 'url': 'https://www.walmart.ca/en/search?q=',
+            'url': '',
+            'regex': r'N°(\d+)',
+            'multi_items': False,
+        },
+        # Add more shops and their respective information here
+    }
 
+    regex_price = r'(\d+\.?\d*) ?([€$])|([€$]) ?(\d+\.?\d*)'
+
+
+    # def app(img, selected_shop):
+    #     if img is None:
+    #         raise gr.Error("No file uploaded. Please upload an image.")
+    #
+    #     # Check if the uploaded file is a PDF
+    #     if selected_shop is None:
+    #         raise gr.Error("Please select a shop.")
+    #
+    #     try:
+    #         image_annotated = section(img)
+    #         pass
+    #     except Exception as e:
+    #         return gr.Error(f"An error occurred: {str(e)}")
+    #
+    #     return image_annotated, hotspots_sku_links[selected_shop]
+
+    def build_request(query, start=1, num=10, date_restrict=None, **params):
+        request = {
+            'key': GOOGLE_API_KEY,
+            'q': query,
+            'cx': GOOGLE_CSE_ID,
+            # 'start': start,
+            # 'num': num,
+            # 'dateRestrict': date_restrict,
+        }
+        request.update(params)
+        return request
+
+
+    def make_request(request):
+        response = requests.get('https://www.googleapis.com/customsearch/v1', params=request)
+        # print(f"Status Code: {response.status_code}")  # Print status code for debugging
+        if response.status_code != 200:
+            raise Exception("Google Custom Search request failed with status code {}".format(response.status_code))
+
+        search_results = response.json()
+        # print(f"Response JSON: {search_results}")  # Print raw response for debugging
+        if 'items' in search_results and len(search_results['items']) > 0:
+            first_result_url = search_results['items'][0]['link']
+            return first_result_url
+        else:
+            return None
 
-with gr.Blocks() as demo:
 
     with gr.Row():
         selected_shop = gr.Dropdown(shops.keys(), label="Shop")