Upload 2 files

app.py

@@ -0,0 +1,121 @@
+import re
+from collections import defaultdict
+import gradio as gr
+import pandas as pd
+
+def create_recommendation_system(pairs):
+    sets_by_length = defaultdict(list)
+
+    for pair in pairs:
+        sets_by_length[len(pair)].append(set(pair))
+
+    sorted_keys = sorted(sets_by_length.keys(), reverse=True)
+
+    def recommend(restaurant):
+        recommended = set()
+        for length in sorted_keys:
+            for rest_set in sets_by_length[length]:
+                if restaurant in rest_set:
+                    recommended.update(rest_set - {restaurant})
+        return list(recommended)
+
+    return recommend
+
+def recommend(input_string):
+    f = open("task1_output_n_c1_sup4 (2).txt", "r")
+    data = f.read()
+    data = data.split("Frequent Itemsets:")
+
+    clean_string = re.sub(r'\n', '', data[1])
+    pattern = r"\([^\)]+\)"
+
+    extracted_tuples = re.findall(pattern, clean_string.replace("'", ""))
+
+    tuple_list = [tuple(s.strip('()').split(', ')) for s in extracted_tuples]
+
+    recommendation_system = create_recommendation_system(tuple_list)
+
+    output_list = recommendation_system(input_string)
+
+    images_labels = [("https://logowik.com/content/uploads/images/aw-restaurants5299.jpg", "A&W Restaurant"),
+    ("https://inspirebrands.com/wp-content/uploads/2017/10/Arbys.jpg","Arbys"),
+    ("https://www.shutterstock.com/image-vector/vinnytsia-ukraine-october-8-2023-600nw-2372329457.jpg","Burger King"),
+    ("https://static.wixstatic.com/media/cc8696_2f0a91e1d9d245efa98b100adfcdd887~mv2.png/v1/crop/x_0,y_0,w_242,h_71/fill/w_339,h_94,al_c,lg_1,q_85,enc_auto/logo.png","California Kitchen"),
+    ("https://logos-world.net/wp-content/uploads/2022/11/Carls-Jr.-Logo-500x281.png","Carls Jr."),
+    ("https://upload.wikimedia.org/wikipedia/commons/thumb/0/02/Chick-fil-A_Logo.svg/873px-Chick-fil-A_Logo.svg.png","Chick-Fil-A"),
+    ("https://upload.wikimedia.org/wikipedia/en/thumb/3/38/CECLogo2019.svg/330px-CECLogo2019.svg.png","Chuck E. Cheese"),
+    ("https://upload.wikimedia.org/wikipedia/commons/thumb/c/cc/Culver%27s_logo.svg/558px-Culver%27s_logo.svg.png","Culvers"),
+    ("https://upload.wikimedia.org/wikipedia/commons/thumb/a/ae/Dairy_Queen_logo.svg/300px-Dairy_Queen_logo.svg.png", "Dairy Queen"),
+    ("https://logowik.com/content/uploads/images/493_dominospizza.jpg", "Dominos Pizza"),
+    ("https://logos-world.net/wp-content/uploads/2023/03/Five-Guys-Logo-500x281.png", "Five Guys"),
+    ("https://entrackr.com/storage/2023/06/Good-Flippin.jpg", "Good Flippin Burger"),
+    ("https://logowik.com/content/uploads/images/hardees4024.jpg", "Hardees"),
+    ("https://logos-marques.com/wp-content/uploads/2023/04/In-N-Out-Burger-Logo-768x432.png", "In-N-Out"),
+    ("https://logos-world.net/wp-content/uploads/2022/08/Jack-in-the-Box-Logo-500x281.png", "Jack in the Box"),
+    ("https://1000logos.net/wp-content/uploads/2021/05/Jollibee-logo-768x432.png", "Jollibee"),
+    ("https://logowik.com/content/uploads/images/674_kfc.jpg", "KFC"),
+    ("https://visitguernseycounty.com/wp-content/uploads/2022/05/Little-Caesars-Logo-1024x576.jpg", "Little Caesars"),
+    ("https://logowik.com/content/uploads/images/mcdonalds-icon.jpg", "McDonalds"),
+    ("https://images.squarespace-cdn.com/content/v1/53a47e51e4b0e78ae9ed2e97/4b8df8ab-662a-4381-92a6-97839099d5a7/Papa-Johns-logo.jpg", "Papa Johns"),
+    ("https://cdn.nwe.io/files/x/d5/cc/c9810c9ad13188a3e0b2bc6577d7.jpg", "Pizza Inn"),
+    ("https://logowik.com/content/uploads/images/294_pizza_hut_new_logo.jpg", "Pizza hut"),
+    ("https://upload.wikimedia.org/wikipedia/commons/a/a6/PizzaExpress_Logo.jpg", "PizzaExpress"),
+    ("https://images.squarespace-cdn.com/content/v1/53a47e51e4b0e78ae9ed2e97/be0aea01-84cc-4b89-a5e8-314ed394dc3c/Popeyes+logo.jpg", "Popeyes"),
+    ("https://upload.wikimedia.org/wikipedia/commons/d/d9/Original_Round_Table_Pizza_Logo.jpg", "Round Table Pizza"),
+    ("https://logowik.com/content/uploads/images/328_sbarro.jpg", "Sbarro"),
+    ("https://1000logos.net/wp-content/uploads/2023/04/Shake-Shack-Logo-768x432.png", "Shake Shack"),
+    ("https://upload.wikimedia.org/wikipedia/en/thumb/6/6d/Shakey%27s_US_logo.svg/330px-Shakey%27s_US_logo.svg.png", "Shakeys Pizza"),
+    ("https://upload.wikimedia.org/wikipedia/commons/thumb/f/ff/SONIC_New_Logo_2020.svg/1199px-SONIC_New_Logo_2020.svg.png", "Sonic"),
+    ("https://images.getbento.com/accounts/236fb3743b9522eafb90c6d2d20b8115/media/accounts/media/V1G5LTCrSDqyixbNUDpr_logo-tasty-burger.png", "Tasty burger"),
+    ("https://cdn.worldvectorlogo.com/logos/uno-pizzeria.svg", "Uno Pizzeria"),
+    ("https://upload.wikimedia.org/wikipedia/commons/thumb/2/28/Wahlburgers_Logo.svg/768px-Wahlburgers_Logo.svg.png", "Wahlburgers"),
+    ("https://logowik.com/content/uploads/images/866_wendys.jpg", "Wendys"),
+    ("https://logowik.com/content/uploads/images/whataburger8433.jpg", "Whataburger"),
+    ("https://encrypted-tbn2.gstatic.com/images?q=tbn:ANd9GcTsUZPMFzUmACfaeSrMSZsgGn5gnbXNyYRPxIrS1PfCrFWI9QJhrlkmdG9IhFbi", "White Castle")]
+
+    output = []
+    for i in images_labels:
+        if(i[1] in output_list):
+          output.append(i)
+
+    return output
+
+sorted_unique_restaurants = [
+ 'A&W Restaurant',
+ 'Arbys',
+ 'Burger King',
+ 'California Kitchen',
+ 'Carls Jr.',
+ 'Chick-Fil-A',
+ 'Chuck E. Cheese',
+ 'Culvers',
+ 'Dairy Queen',
+ 'Dominos Pizza',
+ 'Five Guys',
+ 'Good Flippin Burger',
+ 'Hardees',
+ 'In-N-Out',
+ 'Jack in the Box',
+ 'Jollibee',
+ 'KFC',
+ 'Little Caesars',
+ 'McDonalds',
+ 'Papa Johns',
+ 'Pizza Inn',
+ 'Pizza hut',
+ 'PizzaExpress',
+ 'Popeyes',
+ 'Round Table Pizza',
+ 'Sbarro',
+ 'Shake Shack',
+ 'Shakeys Pizza',
+ 'Sonic',
+ 'Tasty burger',
+ 'Uno Pizzeria',
+ 'Wahlburgers',
+ 'Wendys',
+ 'Whataburger',
+ 'White Castle']
+
+io = gr.Interface(fn=recommend, inputs=gr.Dropdown(choices=sorted_unique_restaurants, value="Burger King"), outputs=gr.Gallery(label="Generated images", show_label=False, elem_id="gallery", columns=[4], object_fit="contain", height="auto"), title='Restaurant Recommender System', description='Find your ideal restaurant')  
+io.queue().launch(share = True, debug = True)