Update app.py

app.py

@@ -1,22 +1,11 @@
 import gradio as gr
 import requests
 from bs4 import BeautifulSoup
-from selenium import webdriver
-from selenium.webdriver.chrome.service import Service
-from webdriver_manager.chrome import ChromeDriverManager
-from selenium.webdriver.common.by import By
-from selenium.webdriver.common.keys import Keys
-import time
 from PIL import Image
 from transformers import ViltProcessor, ViltForQuestionAnswering
 import torch
 import torch.nn.functional as F
 
-# Selenium WebDriver setup
-options = webdriver.ChromeOptions()
-options.add_argument("--headless")  # Futás háttérben
-driver = webdriver.Chrome(service=Service(ChromeDriverManager().install()), options=options)
-
 # ViLT Model betöltése
 model_path = "dandelin/vilt-b32-finetuned-vqa"  # Hugging Face modell
 device = "cuda" if torch.cuda.is_available() else "cpu"
@@ -27,25 +16,30 @@ processor = ViltProcessor.from_pretrained(model_path)
 # Keresés a Roblox katalógusban
 def search_roblox(character_name):
     url = f"https://www.roblox.com/catalog?Category=3&salesTypeFilter=1"
-    driver.get(url)
-    time.sleep(3)
-
-    search_box = driver.find_element(By.ID, "avatar-shop-keyword-input")
-    search_box.clear()
-    search_box.send_keys(character_name)
-    search_box.send_keys(Keys.RETURN)
-    time.sleep(5)
-
-    items = driver.find_elements(By.XPATH, "//div[contains(@class, 'catalog-item-container')]")
+    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.36'}
+    response = requests.get(url, headers=headers)
+    
+    if response.status_code != 200:
+        return None, "Failed to fetch data", 0
+    
+    soup = BeautifulSoup(response.text, "html.parser")
+    
+    # Az elemzéshez szükséges adatokat a megfelelő HTML elemekből szűrjük
+    items = soup.find_all('div', {'class': 'catalog-item-container'})
     
     best_item = None
     best_score = 0
 
     for item in items[:5]:  # Csak az első 5 találatot nézzük
         try:
-            img_url = item.find_element(By.XPATH, ".//img").get_attribute("src")
-            name = item.find_element(By.XPATH, ".//div[contains(@class, 'item-card-name')]").text
-            link = item.find_element(By.XPATH, ".//a[contains(@class, 'item-card-link')]").get_attribute("href")
+            img_tag = item.find('img')
+            img_url = img_tag['src'] if img_tag else None
+            name = item.find('div', {'class': 'item-card-name'}).text.strip()
+            link_tag = item.find('a', {'class': 'item-card-link'})
+            link = link_tag['href'] if link_tag else None
+            
+            if not img_url or not link:
+                continue
 
             image = Image.open(requests.get(img_url, stream=True).raw).convert("RGB")