Spaces:
Runtime error
Runtime error
Create app.py
Browse files
app.py
ADDED
|
@@ -0,0 +1,45 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# App v3.0
|
| 2 |
+
import pandas as pd
|
| 3 |
+
import pickle
|
| 4 |
+
import gradio as gr
|
| 5 |
+
|
| 6 |
+
with open(r'database\vect_index.pickle', 'rb') as file:
|
| 7 |
+
vect_index = pickle.load(file)
|
| 8 |
+
with open(r'database\vect_values.pickle', 'rb') as file:
|
| 9 |
+
vect_values = pickle.load(file)
|
| 10 |
+
|
| 11 |
+
df = pd.read_excel('database\perfume_database_cleaned.xlsx')
|
| 12 |
+
brand_options = list(df['brand'].unique())
|
| 13 |
+
perfume_options = list(df['perfume'].unique())
|
| 14 |
+
|
| 15 |
+
def perfume_change(brand):
|
| 16 |
+
names = list(df.query(f"brand=='{brand}'")['perfume'])
|
| 17 |
+
return gr.update(choices=names, value=None)
|
| 18 |
+
|
| 19 |
+
def filter_by_perfume(selected_brand, selected_perfume):
|
| 20 |
+
perfume_index = df.query(f"brand=='{selected_brand}' & perfume=='{selected_perfume}'").index[0]
|
| 21 |
+
df_filter = df.iloc[vect_index[perfume_index], :].reset_index(drop=True)
|
| 22 |
+
df_filter['similarity'] = vect_values[perfume_index]
|
| 23 |
+
df_filter['similarity'] = df_filter['similarity'].map("{:.2%}".format)
|
| 24 |
+
df_filter['notes'] = df_filter['notes'].str.split(',').apply(sorted, 1).str.join(',').str.strip(',')
|
| 25 |
+
df_filter = df_filter[['brand', 'perfume', 'similarity', 'notes']]
|
| 26 |
+
return df_filter
|
| 27 |
+
|
| 28 |
+
with gr.Blocks(theme='freddyaboulton/dracula_revamped') as demo:
|
| 29 |
+
gr.Markdown(
|
| 30 |
+
"""
|
| 31 |
+
# Perfume Recommendation
|
| 32 |
+
Select brand and perfume below to find top 5 most similar perfumes based on notes
|
| 33 |
+
""")
|
| 34 |
+
|
| 35 |
+
with gr.Row():
|
| 36 |
+
brand_dropdown = gr.Dropdown(choices=brand_options, value='Hugo Boss', label="Brand")
|
| 37 |
+
perfume_dropdown = gr.Dropdown(choices=perfume_options, value='Boss Soul', label="Perfume Name")
|
| 38 |
+
brand_dropdown.change(fn=perfume_change, inputs=[brand_dropdown], outputs=[perfume_dropdown])
|
| 39 |
+
btn = gr.Button(value="Search", scale=.1)
|
| 40 |
+
|
| 41 |
+
with gr.Row():
|
| 42 |
+
output_df = gr.outputs.Dataframe(type='pandas')
|
| 43 |
+
btn.click(filter_by_perfume, inputs=[brand_dropdown, perfume_dropdown], outputs=[output_df])
|
| 44 |
+
|
| 45 |
+
demo.launch()
|