Update app.py

app.py

@@ -3,28 +3,39 @@ import requests
 import time
 import random
 import uuid
+import matplotlib.pyplot as plt
+import pandas as pd
 
-# Список User-Agent для імітації різних пристроїв
+# Розширений список 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 (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"
+# Генерація випадкових 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)  # Випадковий User-Agent для імітації різних користувачів
+        user_agent = random.choice(USER_AGENTS)
+        ip_address = generate_random_ip()
 
         headers = {
             "User-Agent": user_agent
         }
 
-        # Дані для відправки (можна налаштувати під ваші потреби)
         payload = {
             "data": [
                 f"test_input_{unique_id}",
@@ -32,22 +43,19 @@ def send_click_to_url(url):
             ]
         }
 
-        # Відправка POST-запиту з заголовками
         response = requests.post(url, json=payload, headers=headers)
 
         if response.status_code == 200:
-            return {"status": "Успішно", "user_agent": user_agent, "ip": f"192.168.{random.randint(0,255)}.{random.randint(0,255)}"}
+            return {"status": "Успішно", "user_agent": user_agent, "ip": ip_address, "timestamp": timestamp}
         else:
-            return {"status": f"Помилка {response.status_code}", "user_agent": user_agent, "ip": f"192.168.{random.randint(0,255)}.{random.randint(0,255)}"}
+            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": f"192.168.{random.randint(0,255)}.{random.randint(0,255)}"}
+        return {"status": "Помилка з'єднання", "user_agent": user_agent, "ip": ip_address, "timestamp": timestamp}
 
 # Функція для симуляції трафіку
 def simulate_traffic(url, interval=1, clicks=5):
-    successful_clicks = 0
-    failed_clicks = 0
-    details = []
+    results = []
     start_time = time.time()
 
     if not url.startswith("http"):
@@ -55,38 +63,68 @@ def simulate_traffic(url, interval=1, clicks=5):
 
     for i in range(clicks):
         result = send_click_to_url(url)
-        if result["status"] == "Успішно":
-            successful_clicks += 1
-        else:
-            failed_clicks += 1
-
-        details.append(f"Клік #{i + 1}: {result['status']} | IP: {result['ip']} | User-Agent: {result['user_agent']}")
+        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"Успішні кліки: {successful_clicks}\n"
-        f"Неуспішні кліки: {failed_clicks}\n"
+        f"Успішні кліки: {len(df[df['status'] == 'Успішно'])}\n"
+        f"Неуспішні кліки: {len(df[df['status'] != 'Успішно'])}\n"
         f"Загальний час симуляції: {total_time} секунд\n"
     )
 
-    return summary + "\n\nДеталі кліків:\n" + "\n".join(details)
+    return summary + "\n\nДля перегляду детальної статистики перегляньте файл 'traffic_log.csv'."
 
-# Інтерфейс Gradio
+# Візуалізація статистики
+def visualize_traffic():
+    try:
+        df = pd.read_csv('traffic_log.csv')
+        
+        # Графік успішних і неуспішних кліків
+        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)
 
-interface = gr.Interface(
-    fn=process_url,
-    inputs=[
-        gr.Textbox(label="Введіть URL"),
-        gr.Slider(0.5, 5, step=0.5, label="Інтервал між кліками (секунди)"),
-        gr.Slider(1, 20, step=1, label="Кількість кліків")
-    ],
-    outputs=gr.Textbox(label="Результати симуляції")
-)
+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="Кількість кліків")
+    
+    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="Графік трафіку")
+
+    visualize_button.click(show_visualization, outputs=image_output)
 
 if __name__ == "__main__":
     interface.launch()