play.py fix

1 files changed, 5 insertions, 53 deletions

diff --git a/cogs/downloader.py b/cogs/downloader.py
index c35373f..5fb3d1c 100644
--- a/cogs/downloader.py
+++ b/cogs/downloader.py
@@ -1,34 +1,19 @@
-import base64
 import os
-import os.path
 import tempfile
+from pathlib import Path
 
 import discord
-import requests
 import yt_dlp
 from discord.ext import commands
 from dotenv import load_dotenv
 
 # Load environment variables
 load_dotenv()
-PIXELDRAIN_API_KEY = os.getenv('PIXELDRAIN_API_KEY')
-
-
-def _get_auth_headers():
-    """Generate authentication headers for Pixeldrain API"""
-    if not PIXELDRAIN_API_KEY:
-        raise Exception("Pixeldrain API key not found in environment variables")
-
-    auth_string = f":{PIXELDRAIN_API_KEY}"
-    auth_bytes = auth_string.encode('ascii')
-    base64_auth = base64.b64encode(auth_bytes).decode('ascii')
-    return {"Authorization": f"Basic {base64_auth}"}
 
 
 class Downloader(commands.Cog):
     def __init__(self, bot):
         self.bot = bot
-        self.pixeldrain_url = "https://pixeldrain.com/api/file"
 
     @commands.slash_command(
         name="download",
@@ -54,47 +39,14 @@ class Downloader(commands.Cog):
                     info = ydl.extract_info(url, download=True)
                     actual_filename = ydl.prepare_filename(info)
 
-                await ctx.followup.send("Video downloaded successfully, Uploading to pixeldrain...")
-                # upload to pixeldrain
                 try:
-                    with open(actual_filename, 'rb') as f:
-                        files = {'file': f}
-                        response = requests.post(
-                            self.pixeldrain_url,
-                            files=files,
-                            headers={
-                                "Authorization": f"Basic {base64.b64encode((':' + PIXELDRAIN_API_KEY).encode()).decode()}"}
-                        )
-
-                        try:
-                            response_data = response.json()
-                            print("Pixeldrain response:", response_data)  # Debug log
-
-                            if response.status_code == 201:  # 201 means successful upload
-                                file_id = response_data.get('id')
-                                if file_id:
-                                    file_link = f"https://pixeldrain.com/u/{file_id}"
-                                    await ctx.followup.send(f"oto twoje videło\n {file_link}")
-                                else:
-                                    await ctx.followup.send(
-                                        f"⚠️ Upload successful but no file ID returned. Response: {response_data}")
-                            else:
-                                error_msg = response_data.get('error', 'Unknown error')
-                                if response.status_code == 401:
-                                    error_msg = "Invalid API key. Please check your PIXELDRAIN_API_KEY in .env file."
-                                await ctx.followup.send(f"❌ Upload failed (Status {response.status_code}): {error_msg}")
-                        except ValueError as e:
-                            # If the response is not JSON
-                            error_msg = f"Failed to parse server response: {str(e)}"
-                            if response.text:
-                                error_msg += f"\nResponse: {response.text[:200]}"
-                            await ctx.followup.send(f"❌ {error_msg}")
+                    file = discord.File(actual_filename)
+                    await ctx.followup.send("Pobran", file=file)
                 except Exception as e:
-                    await ctx.followup.send(f"Error: {str(e)}")
-                # No need for explicit cleanup as a tempfile.TemporaryDirectory handles it automatically
+                    await ctx.followup.send(f":(\n{str(e)}")
 
         except Exception as e:
-            await ctx.followup.send(f"Error: {str(e)}")
+            await ctx.followup.send(f":(\n{str(e)}")
 
 
 def setup(bot):