Update app.py

app.py

@@ -1,92 +1,100 @@
 import gradio as gr
-import numpy as np
-from PIL import Image
+import requests
+from bs4 import BeautifulSoup
+import pandas as pd
+from selenium import webdriver
+from selenium.webdriver.chrome.options import Options
+import csv
 import random
+import time
+import os
 
-# Game state variables
-ship_position = 50  # Ship's horizontal position (0-100)
-enemies = []  # List of enemy positions
-score = 0
+# List of user agents to avoid bot detection
+USER_AGENTS = [
+    "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36",
+    "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.114 Safari/537.36",
+    "Mozilla/5.0 (X11; Ubuntu; Linux x86_64; rv:89.0) Gecko/20100101 Firefox/89.0"
+]
 
-# Function to generate a simple game screen as an image
-def create_game_screen(ship_pos, enemy_list, score):
-    # Create a blank 200x200 image (black background)
-    img = np.zeros((200, 200, 3), dtype=np.uint8)
-    
-    # Draw ship (white rectangle) at the bottom
-    ship_y = 180
-    img[ship_y:ship_y+10, max(0, ship_pos-5):min(200, ship_pos+5)] = [255, 255, 255]
-    
-    # Draw enemies (red squares)
-    for enemy_x, enemy_y in enemy_list:
-        img[max(0, enemy_y-5):min(200, enemy_y+5), max(0, enemy_x-5):min(200, enemy_x+5)] = [255, 0, 0]
-    
-    # Convert to PIL Image for Gradio
-    return Image.fromarray(img), f"Score: {score}"
+# Function to initialize Selenium driver (headless)
+def get_driver():
+    chrome_options = Options()
+    chrome_options.add_argument("--headless")  # Run in headless mode
+    chrome_options.add_argument("--no-sandbox")
+    chrome_options.add_argument("--disable-dev-shm-usage")
+    driver = webdriver.Chrome(options=chrome_options)
+    return driver
 
-# Function to update game state based on user action
-def update_game(action):
-    global ship_position, enemies, score
-    
-    # Move ship
-    if action == "Move Left" and ship_position > 10:
-        ship_position -= 10
-    elif action == "Move Right" and ship_position < 190:
-        ship_position += 10
-    
-    # Shooting logic
-    elif action == "Shoot":
-        for i, (enemy_x, enemy_y) in enumerate(enemies[:]):
-            if abs(enemy_x - ship_position) < 15 and enemy_y > 150:  # Rough hit detection
-                enemies.pop(i)
-                score += 10
-                break
+# Function to scrape Flipkart laptop data
+def scrape_flipkart(url):
+    try:
+        # Set up Selenium driver
+        driver = get_driver()
+        headers = {"User-Agent": random.choice(USER_AGENTS)}
+        
+        # Load the page
+        driver.get(url)
+        time.sleep(5)  # Wait for JavaScript to load content
+        
+        # Get page source and parse with BeautifulSoup
+        soup = BeautifulSoup(driver.page_source, "html.parser")
+        driver.quit()  # Close the driver
+        
+        # Lists to store scraped data
+        products = []
+        prices = []
+        ratings = []
+        
+        # Find laptop items (adjust class names based on Flipkart's HTML structure)
+        items = soup.find_all("div", class_="_1AtVbE")  # Parent div for each product
+        for item in items:
+            # Product name
+            name_tag = item.find("div", class_="_4rR01T")
+            name = name_tag.text.strip() if name_tag else "N/A"
+            
+            # Price
+            price_tag = item.find("div", class_="_30jeq3")
+            price = price_tag.text.strip() if price_tag else "N/A"
+            
+            # Rating
+            rating_tag = item.find("div", class_="_3LWZlK")
+            rating = rating_tag.text.strip() if rating_tag else "N/A"
+            
+            if name != "N/A":  # Only append valid entries
+                products.append(name)
+                prices.append(price)
+                ratings.append(rating)
+        
+        # Create DataFrame
+        df = pd.DataFrame({
+            "Product Name": products,
+            "Price": prices,
+            "Rating": ratings
+        })
+        
+        # Save to CSV
+        csv_path = "flipkart_laptops.csv"
+        df.to_csv(csv_path, index=False, encoding="utf-8")
+        
+        return f"Scraped {len(products)} laptops successfully!", csv_path
     
-    # Spawn new enemies randomly
-    if random.random() < 0.3:  # 30% chance to spawn an enemy
-        enemies.append([random.randint(0, 200), 0])
-    
-    # Move enemies down
-    for enemy in enemies[:]:
-        enemy[1] += 10
-        if enemy[1] > 200:  # Remove enemies that go off-screen
-            enemies.remove(enemy)
-    
-    # Generate updated game screen
-    return create_game_screen(ship_position, enemies, score)
-
-# Reset game state
-def reset_game():
-    global ship_position, enemies, score
-    ship_position = 50
-    enemies = []
-    score = 0
-    return create_game_screen(ship_position, enemies, score)
+    except Exception as e:
+        return f"Error: {str(e)}", None
 
 # Gradio interface
-with gr.Blocks(title="Blastar 1984 Simulator") as demo:
-    gr.Markdown("# Blastar 1984 Simulator")
-    gr.Markdown("A simplified version of Elon Musk's 1984 game. Move your ship and shoot enemies!")
-    
-    # Display game screen and score
-    output_image = gr.Image(label="Game Screen")
-    output_text = gr.Textbox(label="Score")
-    
-    # Buttons for controls
-    with gr.Row():
-        btn_left = gr.Button("Move Left")
-        btn_right = gr.Button("Move Right")
-        btn_shoot = gr.Button("Shoot")
-        btn_reset = gr.Button("Reset Game")
+with gr.Blocks(title="Flipkart Laptop Scraper") as demo:
+    gr.Markdown("# Flipkart Laptop Scraper")
+    gr.Markdown("Enter a Flipkart laptop category URL to scrape data and download as CSV.")
     
-    # Event handlers
-    btn_left.click(fn=update_game, inputs=gr.State(value="Move Left"), outputs=[output_image, output_text])
-    btn_right.click(fn=update_game, inputs=gr.State(value="Move Right"), outputs=[output_image, output_text])
-    btn_shoot.click(fn=update_game, inputs=gr.State(value="Shoot"), outputs=[output_image, output_text])
-    btn_reset.click(fn=reset_game, inputs=None, outputs=[output_image, output_text])
+    url_input = gr.Textbox(label="Flipkart URL", placeholder="e.g., https://www.flipkart.com/laptops/pr?sid=6bo,b5g")
+    scrape_btn = gr.Button("Scrape Data")
+    output_text = gr.Textbox(label="Status")
+    output_file = gr.File(label="Download CSV")
     
-    # Initial state
-    demo.load(fn=reset_game, inputs=None, outputs=[output_image, output_text])
+    scrape_btn.click(
+        fn=scrape_flipkart,
+        inputs=url_input,
+        outputs=[output_text, output_file]
+    )
 
-# Launch the app (Hugging Face handles this automatically)
 demo.launch()