import streamlit as st
import requests
from bs4 import BeautifulSoup

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/96.0.4664.110 Safari/537.36'
    }
    response = requests.get(url, headers=headers)

    if response.status_code == 200:
        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:
            return "Информация о VIN-коде не найдена"
    else:
        return "Ошибка при получении информации"

def main():
    # Установка фавикона
    st.set_page_config(page_title="VIN Info Web App", page_icon="🚗")
    
    # Добавляем кликабельную надпись 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()