Update cover.py

cover.py

@@ -9,7 +9,7 @@ from telegram.ext import ContextTypes
 from main import song_cover_pipeline  # Keeping this import from your original main.py
 from webui import download_online_model  # Import the download function
 
-Define paths
+
 
 BASE_DIR = os.path.dirname(os.path.abspath(file))
 output_dir = os.path.join(BASE_DIR, 'song_output')
@@ -18,7 +18,7 @@ Ensure the output directory exists
 
 os.makedirs(output_dir, exist_ok=True)
 
-Start command
+
 
 async def start(update: Update, context: ContextTypes.DEFAULT_TYPE):
 keyboard = [
@@ -29,7 +29,6 @@ keyboard = [
 reply_markup = InlineKeyboardMarkup(keyboard)
 await update.message.reply_text('Welcome to AICoverGen! Choose an option below:', reply_markup=reply_markup)
 
-Button handler
 
 async def button(update: Update, context: ContextTypes.DEFAULT_TYPE):
 query = update.callback_query
@@ -59,7 +58,7 @@ elif query.data == 'help':
     )  
     await query.edit_message_text(text=help_text)
 
-Message handler
+
 
 async def handle_message(update: Update, context: ContextTypes.DEFAULT_TYPE):
 # Check which mode the user is in based on previous button choice
@@ -75,7 +74,7 @@ else:
     # If no mode is selected, ask the user to choose an option  
     await update.message.reply_text("Please choose an option first by clicking 'Generate Song' or 'Download Model'.")
 
-Generate song handler
+
 
 async def generate_song(update: Update, context: ContextTypes.DEFAULT_TYPE):
 song_input = update.message.text
@@ -97,7 +96,6 @@ if os.path.exists(song_output):
 else:  
     await update.message.reply_text(f"An error occurred while generating the song.")
 
-Download model handler with custom name
 
 async def download_model(update: Update, context: ContextTypes.DEFAULT_TYPE):
 model_input = update.message.text
@@ -119,7 +117,6 @@ try:
 except Exception as e:  
     await update.message.reply_text(f"Failed to download the model. Error: {str(e)}")
 
-Main function to run the bot
 
 def main():
 bot_token = "7722898432:AAEfj9s6ubY107SWiF6Uy1IKJFFmsiqY_BA"