Create app.py

app.py

@@ -1,130 +1,45 @@
 import gradio as gr
 import requests
-import time
-import random
-import uuid
-import matplotlib.pyplot as plt
-import pandas as pd
+from bs4 import BeautifulSoup
 
-# Розширений список User-Agent для більш реалістичної імітації трафіку
-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/605.1.15 (KHTML, like Gecko) Version/14.0.3 Safari/605.1.15",
-    "Mozilla/5.0 (Linux; Android 10; SM-G975F) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.114 Mobile Safari/537.36",
-    "Mozilla/5.0 (iPhone; CPU iPhone OS 14_6 like Mac OS X) AppleWebKit/605.1.15 (KHTML, like Gecko) Version/14.0 Mobile/15E148 Safari/604.1",
-    "Mozilla/5.0 (X11; Ubuntu; Linux x86_64; rv:89.0) Gecko/20100101 Firefox/89.0",
-    "Mozilla/5.0 (Windows NT 6.1; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/89.0.4389.82 Safari/537.36",
-    "Mozilla/5.0 (iPad; CPU OS 13_3 like Mac OS X) AppleWebKit/605.1.15 (KHTML, like Gecko) Version/13.0 Mobile/15E148 Safari/604.1",
-    "Mozilla/5.0 (Linux; U; Android 4.4.2; en-us; Nexus 5 Build/KOT49H) AppleWebKit/537.36 (KHTML, like Gecko) Version/4.0 Chrome/30.0.0.0 Mobile Safari/537.36"
-]
-
-# Генерація випадкових IP-адрес для симуляції трафіку
-def generate_random_ip():
-    # Симуляція IP з різних геолокацій (прикладова модель)
-    first_octet = random.choice([23, 45, 67, 89, 123, 156, 189, 210])
-    return f"{first_octet}.{random.randint(0,255)}.{random.randint(0,255)}.{random.randint(0,255)}"
-
-# Функція для відправки унікального кліку
-def send_click_to_url(url):
-    try:
-        unique_id = str(uuid.uuid4())  # Унікальний ID для кожного кліку
-        timestamp = int(time.time())
-        user_agent = random.choice(USER_AGENTS)
-        ip_address = generate_random_ip()
-
-        headers = {
-            "User-Agent": user_agent
-        }
-
-        payload = {
-            "data": [
-                f"test_input_{unique_id}",
-                timestamp
-            ]
-        }
-
-        response = requests.post(url, json=payload, headers=headers)
-
-        if response.status_code == 200:
-            return {"status": "Успішно", "user_agent": user_agent, "ip": ip_address, "timestamp": timestamp}
-        else:
-            return {"status": f"Помилка {response.status_code}", "user_agent": user_agent, "ip": ip_address, "timestamp": timestamp}
-
-    except requests.exceptions.RequestException:
-        return {"status": "Помилка з'єднання", "user_agent": user_agent, "ip": ip_address, "timestamp": timestamp}
-
-# Функція для симуляції трафіку
-def simulate_traffic(url, interval=1, clicks=5):
-    results = []
-    start_time = time.time()
-
-    if not url.startswith("http"):
-        return "Помилка: URL має починатися з 'http://' або 'https://'."
-
-    for i in range(clicks):
-        result = send_click_to_url(url)
-        results.append(result)
-        time.sleep(random.uniform(0, interval))
-
-    total_time = round(time.time() - start_time, 2)
-    
-    # Логування результатів у CSV-файл
-    df = pd.DataFrame(results)
-    df.to_csv('traffic_log.csv', index=False)
-
-    summary = (
-        f"Симуляція завершена!\n"
-        f"Загальна кількість кліків: {clicks}\n"
-        f"Успішні кліки: {len(df[df['status'] == 'Успішно'])}\n"
-        f"Неуспішні кліки: {len(df[df['status'] != 'Успішно'])}\n"
-        f"Загальний час симуляції: {total_time} секунд\n"
-    )
-
-    return summary + "\n\nДля перегляду детальної статистики перегляньте файл 'traffic_log.csv'."
-
-# Візуалізація статистики
-def visualize_traffic():
+def process_url(url, method):
     try:
-        df = pd.read_csv('traffic_log.csv')
+        if method == "Обычный запрос":
+            response = requests.get(url)
+        elif method == "Через прокси":
+            proxies = {
+                "http": "http://your_proxy:port",
+                "https": "http://your_proxy:port"
+            }
+            response = requests.get(url, proxies=proxies)
+        elif method == "Только заголовки":
+            response = requests.head(url)
+            return f"Заголовки:\n{response.headers}"
+
+        response.raise_for_status()
+
+        if method != "Только заголовки":
+            soup = BeautifulSoup(response.content, 'html.parser')
+            page_title = soup.title.string if soup.title else "Нет заголовка"
+            page_preview = soup.prettify()[:500]  # Ограничиваем вывод до 500 символов
+            return f"Заголовок страницы: {page_title}\n\nПредварительный просмотр кода:\n{page_preview}"
         
-        # Графік успішних і неуспішних кліків
-        status_counts = df['status'].value_counts()
-        plt.figure(figsize=(8, 5))
-        plt.bar(status_counts.index, status_counts.values, color=['green', 'red'])
-        plt.title('Статистика кліків')
-        plt.xlabel('Статус')
-        plt.ylabel('Кількість кліків')
-        plt.tight_layout()
-        plt.savefig('traffic_stats.png')
-        return 'traffic_stats.png'
-    except FileNotFoundError:
-        return "Файл з логами не знайдено. Виконайте симуляцію спочатку."
-
-# Інтерфейс Gradio для симуляції і візуалізації
-def process_url(url, interval, clicks):
-    return simulate_traffic(url, interval, clicks)
+    except requests.exceptions.RequestException as e:
+        return f"Ошибка при обработке URL: {e}"
 
-def show_visualization():
-    return visualize_traffic()
-
-interface = gr.Blocks()
-
-with interface:
-    gr.Markdown("### Симуляція трафіку та аналіз статистики")
-    url_input = gr.Textbox(label="Введіть URL")
-    interval_slider = gr.Slider(0.5, 5, step=0.5, label="Інтервал між кліками (секунди)")
-    clicks_slider = gr.Slider(1, 50, step=1, label="Кількість кліків")
+# Создаем интерфейс
+with gr.Blocks() as demo:
+    gr.Markdown("## Введите ссылку для просмотра исходного кода сайта")
     
-    simulate_button = gr.Button("Запустити симуляцію")
-    output_text = gr.Textbox(label="Результати симуляції")
-
-    simulate_button.click(process_url, inputs=[url_input, interval_slider, clicks_slider], outputs=output_text)
-
-    gr.Markdown("### Візуалізація статистики")
-    visualize_button = gr.Button("Показати графік")
-    image_output = gr.Image(label="Графік трафіку")
+    url_input = gr.Textbox(label="Введите URL", placeholder="http://example.com")
+    method_selector = gr.Dropdown(choices=["Обычный запрос", "Через прокси", "Только заголовки"], 
+                                  label="Метод обработки", value="Обычный запрос")
+    
+    submit_button = gr.Button("Отправить")
+    output_text = gr.Textbox(label="Результат", lines=20)
 
-    visualize_button.click(show_visualization, outputs=image_output)
+    # Обработка нажатия кнопки
+    submit_button.click(process_url, inputs=[url_input, method_selector], outputs=output_text)
 
-if __name__ == "__main__":
-    interface.launch()
+# Запуск приложения
+demo.launch(server_name="0.0.0.0", server_port=7860)