Spaces:
Sleeping
Sleeping
| from scraper_tiki import * | |
| from scraper_lazada import * | |
| from scraper_shopee import * | |
| import pandas as pd | |
| import streamlit as st | |
| # #test Tiki | |
| # start_driver() | |
| # DRIVER.get('https://tiki.vn/search?sort=price%2Casc&q=megaduo') | |
| # time.sleep(3) | |
| # products = DRIVER.find_elements(By.CLASS_NAME, 'product-item') | |
| # product = products[2] | |
| # info = get_tiki_product_info_single(product, True) | |
| # print(info) | |
| # # Test Lazada | |
| # start_driver() | |
| # DRIVER.get('https://www.lazada.vn/catalog/?page=1&q=megaduo&sort=priceasc') | |
| # time.sleep(3) | |
| # products = DRIVER.find_elements(By.CLASS_NAME, 'Bm3ON') | |
| # product = products[2] | |
| # info = get_lazada_product_info_single(product, True) | |
| # print(info) | |
| def main(): | |
| st.title("Price Comparison (So Sánh Giá)") | |
| with st.form(key="user_input_form"): | |
| search_product = st.text_input("What would you like to buy? (Bạn muốn mua gì?)") | |
| submit_button = st.form_submit_button(label="Search") | |
| if submit_button: | |
| print('Scraping', search_product) | |
| # search_product = "megaduo" | |
| # search_product = input("Search for what? ") | |
| num_max_page = 1 | |
| extra_info = True | |
| n_products_to_view = 5 # Change this as you like to check more products | |
| col_to_display = ['name', 'price', 'product_url', 'image'] | |
| st.subheader("Shopee") | |
| shopee_data = scrap_shopee(search_product, num_max_page, extra_info) | |
| if shopee_data: | |
| df_shopee = pd.DataFrame(data=shopee_data, columns=shopee_data[0].keys()) | |
| print(df_shopee.head()) | |
| st.write(df_shopee[col_to_display].sort_values(by='price').head(n_products_to_view)) | |
| else: | |
| df_shopee = pd.DataFrame(columns = col_to_display) | |
| st.write("Not found.") | |
| st.subheader("Lazada") | |
| lazada_data = scrap_lazada(search_product, num_max_page, extra_info) | |
| if lazada_data: | |
| df_lazada = pd.DataFrame(data=lazada_data, columns=lazada_data[0].keys()) | |
| print(df_lazada.head()) | |
| st.write(df_lazada[col_to_display].sort_values(by='price').head(n_products_to_view)) | |
| else: | |
| df_lazada = pd.DataFrame(columns = col_to_display) | |
| st.write("Not found.") | |
| st.subheader("Tiki") | |
| tiki_data = scrap_tiki(search_product, num_max_page, extra_info) | |
| if tiki_data: | |
| df_tiki = pd.DataFrame(data=tiki_data, columns=tiki_data[0].keys()) | |
| print(df_tiki.head()) | |
| st.write(df_tiki[col_to_display].sort_values(by='price').head(n_products_to_view)) | |
| else: | |
| df_tiki = pd.DataFrame(columns = col_to_display) | |
| st.write("Not found.") | |
| # Merge the two dataframes | |
| # merged_df = pd.concat([df_tiki, df_lazada, df_shopee]) | |
| merged_df = pd.concat([df_lazada]) | |
| # Sort the merged dataframe by price | |
| sorted_merged_df = merged_df.sort_values(by='price') | |
| print(sorted_merged_df.head(n_products_to_view)) | |
| st.subheader("All sites, sorted by price ascending (Sắp xếp theo giá tăng dần)") | |
| st.write(sorted_merged_df.head(n_products_to_view)) | |
| if __name__ == "__main__": | |
| main() |