Update app.py

app.py

@@ -1,140 +1,196 @@
 import gradio as gr
-import pandas as pd
+import csv
 import requests
-import json
-import random
+import time
 import os
 
-# =========================
-# Random User-Agent Generator
-# =========================
+# File path configuration
+CSV_FILE = 'endpoints.csv'
 
-def generate_random_user_agent(device_type=None, browser_type=None, chrome_versions=[125, 138], firefox_versions=[120, 135]):
-    if not device_type:
-        device_type = random.choice(['android', 'ios', 'windows', 'ubuntu'])
-
-    if not browser_type:
-        browser_type = random.choice(['chrome', 'firefox'])
-
-    if browser_type == 'chrome':
-        major = random.randint(chrome_versions[0], chrome_versions[1])
-        browser_version = f"{major}.{random.randint(0,9)}.{random.randint(1000,9999)}.{random.randint(0,99)}"
+def load_data():
+    """Loads the endpoints from the CSV file."""
+    if not os.path.exists(CSV_FILE):
+        print(f"Error: {CSV_FILE} not found.")
+        return []
+    
+    data = []
+    try:
+        with open(CSV_FILE, 'r', encoding='utf-8') as f:
+            # Assumes headers: TYPE1, TYPE2, URL
+            reader = csv.DictReader(f)
+            for row in reader:
+                # specific cleanup to remove potential whitespace from CSV keys/values
+                cleaned_row = {k.strip(): v.strip() for k, v in row.items() if k}
+                data.append(cleaned_row)
+        return data
+    except Exception as e:
+        print(f"Error reading CSV: {e}")
+        return []
+
+def get_type1_choices():
+    """Extracts unique TYPE1 categories (e.g., profile, video, comments)."""
+    data = load_data()
+    if not data:
+        return []
+    # Set comprehension to get unique values
+    return list(set([item.get('TYPE1') for item in data if item.get('TYPE1')]))
+
+def get_type2_choices(selected_type1):
+    """
+    Extracts TYPE2 options based on selected TYPE1.
+    Adds an 'ALL ...' option at the beginning.
+    """
+    data = load_data()
+    if not data or not selected_type1:
+        return []
+    
+    # Filter data for the selected category
+    choices = [item.get('TYPE2') for item in data if item.get('TYPE1') == selected_type1]
+    unique_choices = list(set(choices))
+    unique_choices.sort()
+    
+    # Add the "ALL" option at the top
+    all_option = f"ALL {selected_type1.upper()} REPORTS"
+    return [all_option] + unique_choices
+
+def execute_reports(type1, type2):
+    """
+    Main logic to send requests to TikTok endpoints.
+    Generator function to stream logs to the UI.
+    """
+    data = load_data()
+    if not data:
+        yield "❌ Error: endpoints.csv not found or empty."
+        return
+
+    targets = []
+    
+    # Construct the "ALL" string to check against
+    all_check_str = f"ALL {type1.upper()} REPORTS"
+
+    # Check if the user selected the "ALL" option
+    if type2 == all_check_str:
+        # Filter all items matching TYPE1
+        targets = [item for item in data if item.get('TYPE1') == type1]
+        yield f"🚀 Preparing to send ALL {len(targets)} reports for category: {type1}...\n"
     else:
-        browser_version = random.randint(firefox_versions[0], firefox_versions[1])
-
-    if device_type == 'android':
-        android_version = random.choice(['10.0','11.0','12.0','13.0','14.0','15.0'])
-        device = random.choice(['Pixel 6','Pixel 7 Pro','SM-G998B','OnePlus 10 Pro'])
-        if browser_type == 'chrome':
-            return f"Mozilla/5.0 (Linux; Android {android_version}; {device}) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/{browser_version} Mobile Safari/537.36"
-        return f"Mozilla/5.0 (Android {android_version}; Mobile; rv:{browser_version}.0) Gecko/{browser_version}.0 Firefox/{browser_version}.0"
-
-    if device_type == 'ios':
-        ios_version = random.choice(['14_0','15_0','16_0'])
-        if browser_type == 'chrome':
-            return f"Mozilla/5.0 (iPhone; CPU iPhone OS {ios_version} like Mac OS X) AppleWebKit/537.36 (KHTML, like Gecko) CriOS/{browser_version} Mobile Safari/604.1"
-        return f"Mozilla/5.0 (iPhone; CPU iPhone OS {ios_version} like Mac OS X) AppleWebKit/605.1.15 (KHTML, like Gecko) FxiOS/{browser_version}.0 Mobile Safari/605.1.15"
-
-    if device_type == 'windows':
-        return f"Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/{browser_version} Safari/537.36"
-
-    return f"Mozilla/5.0 (X11; Ubuntu; Linux x86_64) Gecko Firefox/{browser_version}.0"
-
-
-# =========================
-# Load CSV
-# =========================
-
-CSV_PATH = "endpoints.csv"
-df = pd.read_csv(CSV_PATH)
-
-# =========================
-# Helpers
-# =========================
-
-def get_type1():
-    return sorted(df["TYPE1"].unique())
-
-def get_type2(t1):
-    return sorted(df[df["TYPE1"] == t1]["TYPE2"].unique())
-
-def get_rows(t1, t2):
-    return df[(df["TYPE1"] == t1) & (df["TYPE2"] == t2)]
-
-def build_preview(rows, object_id):
-    if rows.empty:
-        return "No matching endpoints."
-
-    preview = []
-    for i, r in rows.iterrows():
-        preview.append(r.URL.replace("{OBJECT_ID}", object_id))
-    return "\n\n".join(preview)
+        # Filter specific item matching TYPE1 and TYPE2
+        targets = [item for item in data if item.get('TYPE1') == type1 and item.get('TYPE2') == type2]
+        yield f"🚀 Preparing to send specific report: {type2}...\n"
 
-def send_single(row, object_id):
-    url = row.URL.replace("{OBJECT_ID}", object_id)
+    if not targets:
+        yield "❌ No matching endpoints found."
+        return
 
+    # Headers to mimic a real browser (helper to avoid immediate rejection)
     headers = {
-        "User-Agent": generate_random_user_agent(),
-        "Accept": "application/json",
+        "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36",
         "Referer": "https://www.tiktok.com/"
     }
 
-    try:
-        r = requests.get(url, headers=headers, timeout=15)
-        try:
-            return json.dumps(r.json(), indent=2)
-        except:
-            return r.text
-    except Exception as e:
-        return str(e)
-
-# =========================
-# UI
-# =========================
-
-with gr.Blocks(title="TikTok Report Endpoint Explorer") as app:
-    gr.Markdown("""
-    ## TikTok Report Endpoint Explorer  
-    **Research / educational tool**
-
-    **Steps**
-    1. Choose target type  
-    2. Choose report category  
-    3. Enter object ID  
-    4. Preview endpoints  
-    5. Send one request
-    """)
-
-    rows_state = gr.State()
-
-    type1 = gr.Dropdown(label="Target Type", choices=get_type1())
-    type2 = gr.Dropdown(label="Report Category")
-    object_id = gr.Textbox(label="Object ID (video / user / comment)")
+    log_history = ""
+    
+    for i, target in enumerate(targets):
+        url = target.get('URL')
+        report_name = target.get('TYPE2')
+        
+        if not url:
+            continue
 
-    preview = gr.Textbox(label="Preview URLs", lines=8)
-    selection = gr.Dropdown(label="Select Endpoint")
-
-    output = gr.Textbox(label="Response", lines=10)
-
-    type1.change(lambda t: gr.Dropdown.update(choices=get_type2(t)), type1, type2)
-
-    def on_type2(t1, t2, oid):
-        rows = get_rows(t1, t2)
-        options = ["ALL (preview only)"] + [str(i) for i in rows.index]
-        return (
-            gr.Dropdown.update(choices=options),
-            rows,
-            build_preview(rows, oid)
-        )
-
-    type2.change(on_type2, [type1, type2, object_id], [selection, rows_state, preview])
-    object_id.change(on_type2, [type1, type2, object_id], [selection, rows_state, preview])
-
-    def send(sel, rows, oid):
-        if sel == "ALL (preview only)":
-            return "Preview only. Select a single endpoint."
-        return send_single(rows.loc[int(sel)], oid)
-
-    gr.Button("Send Single Request").click(send, [selection, rows_state, object_id], output)
-
-app.launch()
+        try:
+            # Send the request
+            # Note: TikTok endpoints usually ignore GET for actions and require POST, 
+            # but based on your URL structure (params in URL), a GET might trigger the hit 
+            # or the URL is intended to be loaded. 
+            # If these are API endpoints, requests.get is standard.
+            response = requests.get(url, headers=headers)
+            
+            timestamp = time.strftime("%H:%M:%S")
+            
+            if response.status_code == 200:
+                log_entry = f"[{timestamp}] ✅ SENT ({i+1}/{len(targets)}): {report_name}\n"
+            else:
+                log_entry = f"[{timestamp}] ⚠️ STATUS {response.status_code} ({i+1}/{len(targets)}): {report_name}\n"
+            
+            log_history += log_entry
+            yield log_history
+            
+            # Small delay to be polite to the server
+            time.sleep(0.5) 
+            
+        except Exception as e:
+            log_history += f"[{time.strftime('%H:%M:%S')}] ❌ ERROR: {report_name} - {str(e)}\n"
+            yield log_history
+
+    yield log_history + "\n✅ DONE. All tasks completed."
+
+# -------------------------------------------------------------------------
+# UI SETUP
+# -------------------------------------------------------------------------
+
+with gr.Blocks(theme=gr.themes.Soft()) as app:
+    gr.Markdown("# 🤖 TikTok Report Bot Controller")
+    gr.Markdown("Select a category and report type. Data is loaded from `endpoints.csv`.")
+    
+    with gr.Row():
+        with gr.Column(scale=1):
+            # Dropdown 1: Category (profile, video, comments)
+            t1_input = gr.Dropdown(
+                choices=get_type1_choices(), 
+                label="Category (Type 1)",
+                info="Select the target category",
+                value=None, 
+                interactive=True
+            )
+            
+            # Dropdown 2: Specific Report or "ALL"
+            # We initialize it empty; it gets populated when t1_input changes
+            t2_input = gr.Dropdown(
+                choices=[], 
+                label="Report Type (Type 2)",
+                info="Select a specific violation or 'ALL'",
+                interactive=True
+            )
+            
+            send_btn = gr.Button("🚀 Send Report(s)", variant="primary")
+            
+            refresh_btn = gr.Button("🔄 Reload CSV Data", variant="secondary")
+
+        with gr.Column(scale=2):
+            # Output Logs
+            logs_output = gr.Textbox(
+                label="Execution Logs", 
+                placeholder="Waiting for command...", 
+                lines=20,
+                max_lines=20,
+                autoscroll=True
+            )
+
+    # ---------------------------------------------------------------------
+    # EVENT HANDLERS
+    # ---------------------------------------------------------------------
+
+    # 1. When Category changes, update Report Type choices
+    # This fixes the AttributeError by returning a new gr.Dropdown object
+    def update_second_dropdown(choice):
+        new_choices = get_type2_choices(choice)
+        return gr.Dropdown(choices=new_choices, value=new_choices[0] if new_choices else None)
+
+    t1_input.change(fn=update_second_dropdown, inputs=t1_input, outputs=t2_input)
+
+    # 2. Send Button Logic
+    send_btn.click(
+        fn=execute_reports, 
+        inputs=[t1_input, t2_input], 
+        outputs=logs_output
+    )
+    
+    # 3. Refresh Button Logic
+    def refresh_ui():
+        new_t1 = get_type1_choices()
+        return gr.Dropdown(choices=new_t1), gr.Dropdown(choices=[], value=None)
+        
+    refresh_btn.click(fn=refresh_ui, outputs=[t1_input, t2_input])
+
+if __name__ == "__main__":
+    app.launch()