Spaces:
Runtime error
Runtime error
| 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) |