Update app.py

app.py

@@ -1,120 +1,49 @@
-import os
+import streamlit as st
 import requests
 from bs4 import BeautifulSoup
-from urllib.parse import urljoin
-from PIL import Image
-from io import BytesIO
 
-
-def is_excluded_image(img_url):
-    # Проверяем, является ли URL изображения исключенным
-    excluded_keywords = ['flags/ru.png', 'flags/en.png', 'API.v2/Icons/Menu.png', 'mc.yandex.ru', 'Minis']
-    return any(keyword in img_url for keyword in excluded_keywords)
-
-
-def is_excluded_element(link):
-    # Проверяем, является ли элемент исключенным
-    excluded_parents = ['CatSetupText', 'CatSetupDialog', 'ErrorFoundText', 'ErrorFoundDialog']
-
-    parent_id = link.find_parent().get('id', '')
-
-    return parent_id in excluded_parents
-
-def get_car_info(vin):
-    url = f"https://www.ilcats.ru/?vin={vin}&VinAction=Search"
-
-    # Добавляем заголовки User-Agent
+def get_vin_info(vin_code):
+    url = f"https://www.ilcats.ru/?vin={vin_code}&VinAction=Search"
     headers = {
-        'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/58.0.3029.110 Safari/537.3'}
-
-    # Отправляем GET-запрос с заголовками
+        'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/96.0.4664.110 Safari/537.36'
+    }
     response = requests.get(url, headers=headers)
 
     if response.status_code == 200:
-        # Используем lxml в качестве парсера
-        soup = BeautifulSoup(response.text, 'lxml')
-
-        # Находим ссылку на каталог
-        catalog_link = soup.find('a', {'href': lambda x: x and '/vw/?VinAction=Choose' in x})
-
-        if catalog_link:
-            catalog_url = urljoin(url, catalog_link['href'])
-            print(f"Ссылка на каталог: {catalog_url}")
-
-            # Создаем список для хранения абсолютных ссылок и их текстовых подписей
-            links_dict = {}
-
-            while True:
-                # Отправим запрос на страницу каталога и извлечем все ссылки
-                catalog_response = requests.get(catalog_url, headers=headers)
-
-                if catalog_response.status_code == 200:
-                    catalog_soup = BeautifulSoup(catalog_response.text, 'lxml')
-
-                    # Находим все ссылки на странице каталога
-                    catalog_links = catalog_soup.find_all('a', href=True)
-
-                    links_dict.clear()  # Очищаем словарь перед заполнением новыми данными
-
-                    for link in catalog_links:
-                        absolute_link = urljoin(catalog_url, link['href'])
-                        link_text = link.text.strip()
-
-                        # Проверяем, исключен ли текущий элемент
-                        if not is_excluded_element(link):
-                            links_dict[link_text] = absolute_link
-
-                    # Выводим список доступных ссылок с их текстовыми подписями
-                    for i, (text, link) in enumerate(links_dict.items(), 1):
-                        print(f"{i}. {text}")
-
-                    # Запрашиваем у пользователя номер выбранной ссылки
-                    selected_link_number = int(input("Выберите номер ссылки для перехода (0 для выхода): "))
-
-                    if selected_link_number == 0:
-                        break
-
-                    elif 1 <= selected_link_number <= len(links_dict):
-                        selected_link = list(links_dict.values())[selected_link_number - 1]
-                        print(f"Переход по выбранной ссылке: {selected_link}")
-
-                        # Обновляем ссылку каталога для следующего запроса
-                        catalog_url = selected_link
-
-                        # Отобразим изображения на странице
-                        images = catalog_soup.find_all('img', src=True)
-                        for i, img in enumerate(images, 1):
-                            img_url = urljoin(catalog_url, img['src'])
-
-                            # Исключаем определенные изображения
-                            if not is_excluded_image(img_url):
-                                response_img = requests.get(img_url)
-                                if response_img.status_code == 200:
-                                    img_data = BytesIO(response_img.content)
-                                    img = Image.open(img_data)
-
-                                    # Сохраняем изображение на диск
-                                    img_filename = f"image_{i}.png"
-                                    img.save(img_filename)
-
-                                    # Открываем изображение с помощью стандартного приложения
-                                    os.startfile(img_filename)
-
-                    else:
-                        print("Неверный номер ссылки. Попробуйте еще раз.")
-
-                else:
-                    print(f"Ошибка при запросе к каталогу: {catalog_response.status_code}")
-                    break
-
-            print("Выход из программы.")
-
+        soup = BeautifulSoup(response.content, "html.parser")
+        vin_card = soup.find("div", class_="VinCard")
+        if vin_card:
+            # Убираем определенные строки из текста VinCard перед отображением на странице
+            unwanted_texts = ["Открыть каталог", "Список опций", "Свернуть"]
+            cleaned_html = str(vin_card)
+            for text in unwanted_texts:
+                cleaned_html = cleaned_html.replace(text, "")
+
+            return cleaned_html
         else:
-            print("Ссылка на каталог не найдена")
-
+            return "Информация о VIN-коде не найдена"
     else:
-        print(f"Ошибка при запросе: {response.status_code}")
-
-# Пример использования
-vin_code = "wvwzzz3bzxp079085"
-get_car_info(vin_code)
+        return "Ошибка при получении информации"
+
+def main():
+    # Добавляем кликабельную надпись AlfaCross сверху
+    st.markdown("[AlfaCross](https://t.me/alfacross_bot)")
+
+    st.title("VIN Info Web App")
+    
+    vin_code = st.text_input("Введите VIN-код:")
+    
+    if st.button("Получить информацию"):
+        info = get_vin_info(vin_code)
+        st.markdown(info, unsafe_allow_html=True)
+        
+    # Добавляем надпись Starchik снизу
+    st.markdown("[Starchik](https://t.me/Starchik_1)")
+
+# Добавим обработку GET-запросов
+if "vin_code" in st.query_params:
+    vin_code = st.query_params["vin_code"]
+    info = get_vin_info(vin_code)
+    st.markdown(info, unsafe_allow_html=True)
+else:
+    main()