Update Gradio app with multiple files

app.py

@@ -1,32 +1,300 @@
-<think>
-ฉันเห็นว่าแอปได้เริ่มต้นสำเร็จแล้ว! จาก log:
+import gradio as gr
+import threading
+import socket
+import time
+import json
+import requests
+from datetime import datetime
+from typing import Dict, List, Optional
+import logging
+from utils import SOCKS5ProxyServer, ProxyManager
+from config import *
 
-1. SOCKS5 Proxy Server เริ่มต้นสำเร็จที่ 0.0.0.0:1080
-2. Gradio app กำลังทำงานที่ http://0.0.0.0:8080
-3. มี warning ว่า share=True ไม่ได้รองรับใน Hugging Face Spaces แต่นั่นไม่ใช่ปัญหา
+# Create proxy manager
+proxy_manager = ProxyManager()
 
-แอปทำงานได้แล้ว แต่ผู้ใช้อาจจะไม่เห็นเว็บไซต์เพราะ:
-1. อาจจะต้องรอสักครู่ให้แอปเริ่มทำงานเต็มที่
-2. หรือต้องไปที่ URL ที่ถูกต้อง
+def start_proxy_server():
+    """Start proxy server"""
+    success = proxy_manager.start_server()
+    if success:
+        status_html = f"""
+        <div class="status-indicator status-running">
+            <h4>🟢 Status: Running</h4>
+            <p>Proxy Server is running at {proxy_manager.proxy_server.host}:{proxy_manager.proxy_server.port}</p>
+        </div>
+        """
+        return status_html, "✅ Proxy Server started successfully"
+    else:
+        status_html = """
+        <div class="status-indicator status-stopped">
+            <h4>🔴 Status: Stopped</h4>
+            <p>Failed to start Proxy Server - check logs for details</p>
+        </div>
+        """
+        return status_html, "❌ Failed to start Proxy Server"
 
-ให้ฉันอัปเดต Space และให้ข้อมูลเกี่ยวกับสถานะปัจจุบัน
-</think>
+def stop_proxy_server():
+    """Stop proxy server"""
+    proxy_manager.stop_server()
+    status_html = """
+    <div class="status-indicator status-stopped">
+        <h4>🔴 Status: Stopped</h4>
+        <p>Proxy Server stopped</p>
+    </div>
+    """
+    return status_html, "🛑 Proxy Server stopped"
 
-เยี่ยม! 🎉 แอปได้เริ่มต้นสำเร็จแล้ว! 
+def restart_proxy_server():
+    """Restart proxy server"""
+    proxy_manager.stop_server()
+    time.sleep(1)
+    success = proxy_manager.start_server()
+    if success:
+        status_html = f"""
+        <div class="status-indicator status-running">
+            <h4>🟢 Status: Running</h4>
+            <p>Proxy Server restarted successfully at {proxy_manager.proxy_server.host}:{proxy_manager.proxy_server.port}</p>
+        </div>
+        """
+        return status_html, "🔄 Proxy Server restarted successfully"
+    else:
+        status_html = """
+        <div class="status-indicator status-stopped">
+            <h4>🔴 Status: Stopped</h4>
+            <p>Failed to restart Proxy Server</p>
+        </div>
+        """
+        return status_html, "❌ Failed to restart Proxy Server"
 
-จาก log ที่คุณแชร์มา ฉันเห็นว่า:
-- ✅ SOCKS5 Proxy Server เริ่มต้นสำเร็จที่ 0.0.0.0:1080
-- ✅ Gradio web interface กำลังทำงานที่ http://0.0.0.0:8080
-- ⚠️ Warning เรื่อง share=True ไม่สำคัญ เพราะใน Hugging Face Spaces มี public URL อยู่แล้ว
+def get_server_status():
+    """Get current server status"""
+    if proxy_manager.is_running():
+        status_html = f"""
+        <div class="status-indicator status-running">
+            <h4>🟢 Status: Running</h4>
+            <p>Proxy Server is running at {proxy_manager.proxy_server.host}:{proxy_manager.proxy_server.port}</p>
+        </div>
+        """
+    else:
+        status_html = """
+        <div class="status-indicator status-stopped">
+            <h4>🔴 Status: Stopped</h4>
+            <p>Proxy Server is not running</p>
+        </div>
+        """
+    return status_html
 
-**สถานะปัจจุบัน:**
-- Proxy Server: ✅ ทำงานปกติ
-- Web Interface: ✅ พร้อมใช้งาน
-- Public URL: สามารถเข้าถึงได้ผ่าน Hugging Face Spaces
+def get_current_proxy_url():
+    """Get current proxy URL"""
+    if proxy_manager.is_running():
+        return f"socks5://localhost:{PROXY_PORT}"
+    return "Not connected"
 
-หากคุณยังไม่เห็นเว็บไซต์ อาจเป็นเพราะ:
-1. แอปกำลังโหลดข้อมูล (โปรดรอสักครู่)
-2. ลองรีเฟรชหน้าเว็บ
-3. เช็คว่า URL ถูกต้อง
+def get_server_info():
+    """Get server information"""
+    return json.dumps(proxy_manager.get_server_info(), indent=2, ensure_ascii=False)
 
-[Open your Space here](https://huggingface.co/spaces/Taf2023/quick-space-610) หากยังไม่เห็นเว็บไซต์ โปรดแจ้งให้ฉันทราบ!
+def get_usage_stats():
+    """Get usage statistics"""
+    return json.dumps(proxy_manager.get_usage_stats(), indent=2, ensure_ascii=False)
+
+def get_user_stats():
+    """Get user statistics"""
+    return json.dumps(proxy_manager.get_user_stats(), indent=2, ensure_ascii=False)
+
+def get_connection_logs():
+    """Get connection logs"""
+    return json.dumps(proxy_manager.get_recent_logs(), indent=2, ensure_ascii=False)
+
+def refresh_all():
+    """Refresh all data"""
+    return (
+        get_current_proxy_url(),
+        get_server_status(),
+        get_server_info(),
+        get_usage_stats(),
+        get_user_stats(),
+        get_connection_logs()
+    )
+
+def setup_interface():
+    """Create Gradio interface"""
+    
+    # Custom CSS for styling
+    css = """
+    .main-container {
+        max-width: 1200px;
+        margin: auto;
+        padding: 20px;
+    }
+    .status-indicator {
+        padding: 10px;
+        border-radius: 5px;
+        margin: 5px 0;
+    }
+    .status-running {
+        background-color: #d4edda;
+        color: #155724;
+        border: 1px solid #c3e6cb;
+    }
+    .status-stopped {
+        background-color: #f8d7da;
+        color: #721c24;
+        border: 1px solid #f5c6cb;
+    }
+    .info-box {
+        background-color: #e9ecef;
+        padding: 15px;
+        border-radius: 5px;
+        margin: 10px 0;
+        border-left: 4px solid #007bff;
+    }
+    .stats-grid {
+        display: grid;
+        grid-template-columns: repeat(auto-fit, minmax(200px, 1fr));
+        gap: 15px;
+        margin: 20px 0;
+    }
+    .stat-card {
+        background: white;
+        padding: 15px;
+        border-radius: 8px;
+        box-shadow: 0 2px 4px rgba(0,0,0,0.1);
+        border: 1px solid #dee2e6;
+    }
+    .stat-number {
+        font-size: 24px;
+        font-weight: bold;
+        color: #007bff;
+    }
+    .stat-label {
+        color: #6c757d;
+        font-size: 14px;
+    }
+    """
+    
+    with gr.Blocks(css=css, title="SOCKS5 Proxy Dashboard", theme=gr.themes.Soft()) as demo:
+        gr.HTML("""
+        <div class="main-container">
+            <div style="text-align: center; margin-bottom: 30px;">
+                <h1>🔒 SOCKS5 Proxy Server Dashboard</h1>
+                <p style="color: #6c757d;">SOCKS5 Proxy Server Management System with Usage Monitoring</p>
+                <div style="margin-top: 10px;">
+                    <a href="https://huggingface.co/spaces/akhaliq/anycoder" target="_blank" style="text-decoration: none; color: #007bff; font-weight: bold;">Built with anycoder</a>
+                </div>
+            </div>
+        </div>
+        """)
+        
+        # Current status
+        with gr.Row():
+            with gr.Column(scale=1):
+                gr.HTML('<h3>🎮 Controls</h3>')
+                with gr.Row():
+                    start_btn = gr.Button("▶️ Start", variant="primary", size="lg")
+                    stop_btn = gr.Button("⏹️ Stop", variant="stop", size="lg")
+                    restart_btn = gr.Button("🔄 Restart", variant="secondary", size="lg")
+                
+                status_html = gr.HTML()
+                proxy_url_output = gr.Textbox(
+                    label="🌐 Current Proxy URL",
+                    value=get_current_proxy_url,
+                    interactive=False,
+                    elem_classes=["info-box"]
+                )
+                
+                # Show server status
+                server_status_output = gr.HTML()
+                
+            with gr.Column(scale=2):
+                gr.HTML('<h3>📊 Usage Statistics</h3>')
+                with gr.Row():
+                    total_connections = gr.Number(label="📈 Total Connections", value=0)
+                    active_connections = gr.Number(label="🔗 Active Connections", value=0)
+                    bandwidth_used = gr.Number(label="📊 Bandwidth Used (MB)", value=0)
+        
+        # User data
+        with gr.Tab("👥 Users"):
+            user_stats = gr.JSON(
+                label="All User Data",
+                value=get_user_stats
+            )
+        
+        # Server data
+        with gr.Tab("🖥️ Server Info"):
+            server_info = gr.JSON(
+                label="Server Information",
+                value=get_server_info
+            )
+        
+        # Usage statistics
+        with gr.Tab("📊 Statistics"):
+            usage_stats = gr.JSON(
+                label="Usage Statistics",
+                value=get_usage_stats
+            )
+        
+        # Logs
+        with gr.Tab("📝 Logs"):
+            logs = gr.JSON(
+                label="Recent Connection Logs",
+                value=get_connection_logs
+            )
+        
+        # Refresh button
+        with gr.Row():
+            refresh_btn = gr.Button("🔄 Refresh All Data", variant="primary")
+        
+        # Bind events
+        start_btn.click(start_proxy_server, outputs=[server_status_output, proxy_url_output])
+        stop_btn.click(stop_proxy_server, outputs=[server_status_output, proxy_url_output])
+        restart_btn.click(restart_proxy_server, outputs=[server_status_output, proxy_url_output])
+        refresh_btn.click(
+            refresh_all,
+            outputs=[
+                proxy_url_output,
+                server_status_output,
+                server_info,
+                usage_stats,
+                user_stats,
+                logs
+            ]
+        )
+        
+        # Auto update data
+        demo.load(
+            lambda: refresh_all(),
+            outputs=[
+                proxy_url_output,
+                server_status_output,
+                server_info,
+                usage_stats,
+                user_stats,
+                logs
+            ]
+        )
+    
+    return demo
+
+if __name__ == "__main__":
+    # Create interface
+    demo = setup_interface()
+    
+    # Try to start proxy server automatically and show result
+    print("Starting SOCKS5 Proxy Server...")
+    success = proxy_manager.start_server()
+    if success:
+        print(f"✅ SOCKS5 Proxy Server started successfully at {proxy_manager.proxy_server.host}:{proxy_manager.proxy_server.port}")
+    else:
+        print("❌ Failed to start SOCKS5 Proxy Server - please check logs")
+    
+    # Start UI with appropriate settings
+    demo.launch(
+        server_name="0.0.0.0",
+        server_port=8080,
+        show_api=False,
+        share=False,
+        ssr_mode=False,
+        inbrowser=False,
+        debug=False
+    )

config.py

@@ -1,16 +1,16 @@
-# การตั้งค่า SOCKS5 Proxy Server
+# SOCKS5 Proxy Server Configuration
 PROXY_HOST = "0.0.0.0"
 PROXY_PORT = 1080
 
-# การตั้งค่า Web Interface
+# Web Interface Configuration
 WEB_HOST = "0.0.0.0"
 WEB_PORT = 8080
 
-# การตั้งค่า Logging
+# Logging Configuration
 LOG_LEVEL = "INFO"
 LOG_FILE = "proxy.log"
 
-# การตั้งค่าอื่นๆ
+# Other Configuration
 MAX_CONNECTIONS = 100
-CONNECTION_TIMEOUT = 300  # วินาที
+CONNECTION_TIMEOUT = 300  # seconds
 BUFFER_SIZE = 4096  # bytes

utils.py

@@ -29,7 +29,7 @@ class SOCKS5ProxyServer:
         self._lock = threading.Lock()
         
     def start(self):
-        """เริ่มต้น SOCKS5 proxy server"""
+        """Start SOCKS5 proxy server"""
         try:
             self.server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
             self.server_socket.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
@@ -38,34 +38,34 @@ class SOCKS5ProxyServer:
             self.running = True
             self.start_time = datetime.now()
             
-            logging.info(f"SOCKS5 Proxy Server เริ่มต้นที่ {self.host}:{self.port}")
+            logging.info(f"SOCKS5 Proxy Server started at {self.host}:{self.port}")
             
-            # เริ่ม thread สำหรับ accept connections
+            # Start thread for accepting connections
             threading.Thread(target=self._accept_connections, daemon=True).start()
             
             return True
         except Exception as e:
-            logging.error(f"เริ่มต้น proxy server ผิดพลาด: {e}")
+            logging.error(f"Failed to start proxy server: {e}")
             return False
     
     def stop(self):
-        """หยุด SOCKS5 proxy server"""
+        """Stop SOCKS5 proxy server"""
         self.running = False
         if self.server_socket:
             try:
                 self.server_socket.close()
             except:
                 pass
-        logging.info("SOCKS5 Proxy Server หยุดทำงาน")
+        logging.info("SOCKS5 Proxy Server stopped")
     
     def _accept_connections(self):
-        """รับการเชื่อมต่อใหม่"""
+        """Accept new connections"""
         while self.running:
             try:
                 client_socket, client_address = self.server_socket.accept()
                 client_ip, client_port = client_address
                 
-                # บันทึกการเชื่อมต่อ
+                # Record connection
                 with self._lock:
                     self.connections[f"{client_ip}:{client_port}"] = ConnectionInfo(
                         ip=client_ip,
@@ -74,7 +74,7 @@ class SOCKS5ProxyServer:
                     )
                     self.total_connections += 1
                 
-                # สร้าง thread สำหรับจัดการการเชื่อมต่อ
+                # Create thread for handling connection
                 threading.Thread(
                     target=self._handle_client,
                     args=(client_socket, client_ip, client_port),
@@ -83,22 +83,22 @@ class SOCKS5ProxyServer:
                 
             except socket.error:
                 if self.running:
-                    logging.error("Error ในการรับการเชื่อมต่อ")
+                    logging.error("Error accepting connection")
                 break
     
     def _handle_client(self, client_socket, client_ip, client_port):
-        """จัดการการเชื่อมต่อจาก client"""
+        """Handle client connection"""
         try:
             # SOCKS5 handshake
             if not self._handle_socks5_handshake(client_socket):
                 return
             
-            # รับคำขอจาก client
+            # Receive request from client
             request = client_socket.recv(1024)
             if len(request) < 10:
                 return
             
-            # แยกคำขอ SOCKS5
+            # Parse SOCKS5 request
             if request[0] != 0x05:  # Version 5
                 return
             
@@ -113,23 +113,23 @@ class SOCKS5ProxyServer:
                 self._handle_udp_associate(client_socket, client_ip, client_port)
                 
         except Exception as e:
-            logging.error(f"Error ในการจัดการ client {client_ip}:{client_port}: {e}")
+            logging.error(f"Error handling client {client_ip}:{client_port}: {e}")
         finally:
-            # ปิดการเชื่อมต่อ
+            # Close connection
             try:
                 client_socket.close()
             except:
                 pass
             
-            # อัปเดตสถานะ
+            # Update status
             with self._lock:
                 if f"{client_ip}:{client_port}" in self.connections:
                     self.connections[f"{client_ip}:{client_port}"].is_active = False
     
     def _handle_socks5_handshake(self, client_socket):
-        """จัดการ SOCKS5 handshake"""
+        """Handle SOCKS5 handshake"""
         try:
-            # รับ client hello
+            # Receive client hello
             data = client_socket.recv(1024)
             if len(data) < 2:
                 return False
@@ -140,7 +140,7 @@ class SOCKS5ProxyServer:
             if version != 0x05:  # Version 5
                 return False
             
-            # ตอบกลับว่าใช้ no authentication
+            # Respond with no authentication
             response = b'\x05\x00'
             client_socket.send(response)
             return True
@@ -150,7 +150,7 @@ class SOCKS5ProxyServer:
             return False
     
     def _handle_connect(self, client_socket, request, client_ip, client_port):
-        """จัดการ SOCKS5 CONNECT command"""
+        """Handle SOCKS5 CONNECT command"""
         try:
             addr_type = request[3]
             dest_addr = ""
@@ -169,15 +169,15 @@ class SOCKS5ProxyServer:
                 dest_port = struct.unpack('!H', request[20:22])[0]
                 dest_addr = dest_ip
             
-            # เชื่อมต่อไปยังเป้าหมาย
+            # Connect to target
             target_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
             target_socket.connect((dest_addr, dest_port))
             
-            # ตอบกลับว่าสำเร็จ
+            # Send success response
             reply = b'\x05\x00\x00\x01' + socket.inet_aton('0.0.0.0') + b'\x00\x00'
             client_socket.send(reply)
             
-            # เริ่มตัวกลางข้อมูล
+            # Start data relay
             threading.Thread(
                 target=self._relay_data,
                 args=(client_socket, target_socket, client_ip, client_port),
@@ -187,7 +187,7 @@ class SOCKS5ProxyServer:
             logging.info(f"CONNECT: {client_ip}:{client_port} -> {dest_addr}:{dest_port}")
             
         except Exception as e:
-            # ส่ง error response
+            # Send error response
             reply = b'\x05\x05\x00\x01\x00\x00\x00\x00\x00\x00'
             try:
                 client_socket.send(reply)
@@ -196,8 +196,8 @@ class SOCKS5ProxyServer:
             logging.error(f"CONNECT error: {e}")
     
     def _handle_bind(self, client_socket, client_ip, client_port):
-        """จัดการ SOCKS5 BIND command"""
-        # BIND command implementation (สำหรับ incoming connections)
+        """Handle SOCKS5 BIND command"""
+        # BIND command implementation (for incoming connections)
         reply = b'\x05\x07\x00\x01\x00\x00\x00\x00\x00\x00'  # Command not supported
         try:
             client_socket.send(reply)
@@ -205,7 +205,7 @@ class SOCKS5ProxyServer:
             pass
     
     def _handle_udp_associate(self, client_socket, client_ip, client_port):
-        """จัดการ SOCKS5 UDP ASSOCIATE command"""
+        """Handle SOCKS5 UDP ASSOCIATE command"""
         # UDP ASSOCIATE implementation
         reply = b'\x05\x00\x00\x01\x00\x00\x00\x00\x00\x00'
         try:
@@ -214,7 +214,7 @@ class SOCKS5ProxyServer:
             pass
     
     def _relay_data(self, client_socket, target_socket, client_ip, client_port):
-        """ส่งข้อมูลระหว่าง client และ target"""
+        """Relay data between client and target"""
         try:
             while True:
                 try:
@@ -222,10 +222,10 @@ class SOCKS5ProxyServer:
                     if not data:
                         break
                     
-                    # ส่งต่อไปยัง target
+                    # Forward to target
                     target_socket.send(data)
                     
-                    # อัปเดตสถิติ
+                    # Update statistics
                     with self._lock:
                         if f"{client_ip}:{client_port}" in self.connections:
                             conn = self.connections[f"{client_ip}:{client_port}"]
@@ -239,10 +239,10 @@ class SOCKS5ProxyServer:
                     if not data:
                         break
                     
-                    # ส่งต่อกลับไปยัง client
+                    # Forward back to client
                     client_socket.send(data)
                     
-                    # อัปเดตสถิติ
+                    # Update statistics
                     with self._lock:
                         if f"{client_ip}:{client_port}" in self.connections:
                             conn = self.connections[f"{client_ip}:{client_port}"]
@@ -267,86 +267,19 @@ class ProxyManager:
         self.port = port
         
     def start_server(self):
-        """เริ่มต้น proxy server"""
+        """Start proxy server"""
         return self.proxy_server.start()
     
     def stop_server(self):
-        """หยุด proxy server"""
+        """Stop proxy server"""
         self.proxy_server.stop()
     
     def is_running(self):
-        """ตรวจสอบว่า server กำลังทำงานอยู่หรือไม่"""
+        """Check if server is running"""
         return self.proxy_server.running
     
     def get_server_info(self):
-        """ดึงข้อมูล server"""
+        """Get server information"""
         info = {
-            "server_status": "รันอยู่" if self.proxy_server.running else "หยุดทำงาน",
-            "host": self.proxy_server.host,
-            "port": self.proxy_server.port,
-            "version": "SOCKS5",
-            "authentication": "ไม่ต้องยืนยันตัวตน",
-            "start_time": self.proxy_server.start_time.isoformat() if self.proxy_server.start_time else None,
-            "uptime": str(datetime.now() - self.proxy_server.start_time) if self.proxy_server.start_time else "00:00:00"
-        }
-        return info
-    
-    def get_usage_stats(self):
-        """ดึงสถิติการใช้งาน"""
-        total_connections = len([c for c in self.proxy_server.connections.values() if not c.is_active])
-        active_connections = len([c for c in self.proxy_server.connections.values() if c.is_active])
-        
-        total_bytes = sum(c.bytes_sent + c.bytes_received for c in self.proxy_server.connections.values())
-        
-        stats = {
-            "total_connections": total_connections,
-            "active_connections": active_connections,
-            "total_bandwidth_bytes": total_bytes,
-            "total_bandwidth_mb": round(total_bytes / (1024 * 1024), 2),
-            "peak_concurrent": max(active_connections, 0),
-            "average_session_duration": "ไม่มีข้อมูล"
-        }
-        
-        return stats
-    
-    def get_user_stats(self):
-        """ดึงข้อมูลผู้ใช้"""
-        users = []
-        for conn_id, conn in self.proxy_server.connections.items():
-            users.append({
-                "id": conn_id,
-                "ip": conn.ip,
-                "port": conn.port,
-                "connected_since": conn.start_time.isoformat(),
-                "is_active": conn.is_active,
-                "bytes_sent": conn.bytes_sent,
-                "bytes_received": conn.bytes_received,
-                "total_bytes": conn.bytes_sent + conn.bytes_received,
-                "session_duration": str(datetime.now() - conn.start_time)
-            })
-        
-        return {
-            "total_users": len(users),
-            "active_users": len([u for u in users if u["is_active"]]),
-            "users": users
-        }
-    
-    def get_recent_logs(self):
-        """ดึงล็อกการเชื่อมต่อล่าสุด"""
-        logs = []
-        for conn_id, conn in self.proxy_server.connections.items():
-            logs.append({
-                "timestamp": conn.start_time.isoformat(),
-                "client": f"{conn.ip}:{conn.port}",
-                "status": "เชื่อมต่ออยู่" if conn.is_active else "เสร็จสิ้น",
-                "bytes_transferred": conn.bytes_sent + conn.bytes_received,
-                "session_duration": str(datetime.now() - conn.start_time)
-            })
-        
-        # เรียงตามเวลาใหม่สุดก่อน
-        logs.sort(key=lambda x: x["timestamp"], reverse=True)
-        
-        return {
-            "recent_connections": logs[:50],  # แสดง 50 รายการล่าสุด
-            "total_log_entries": len(logs)
-        }
+            "server_status": "Running" if self.proxy_server.running else "Stopped",
+            "host": self