Update app3.py

app3.py

@@ -1,7 +1,7 @@
 #!/usr/bin/env python3
 """
-Discord Bot for Voiceover Generation
-Reads prompts.csv from server and generates voiceovers
+Discord Bot - Reads CSV from #general channel
+CSV file is uploaded in #general, bot reads it from there
 """
 
 import discord
@@ -9,16 +9,19 @@ from discord.ext import commands
 import asyncio
 import edge_tts
 import csv
-import os
+import io
 import tempfile
-from pathlib import Path
+import random
 
 # ============================================================
 # CONFIGURATION
 # ============================================================
 TOKEN = "MTQ2ODk2MDIzMjg4MDQ3MjIwNg.GkZcck.IotGRnly_kDwVwUkK9MZ_KXaD6TYI-4lGFKepI"
 
-# Voice styles for Edge TTS
+# Channel name where CSV is uploaded
+CSV_CHANNEL_NAME = "general"  # #general channel
+
+# Voice styles
 VOICE_STYLES = {
     "1": {"name": "Professional Female", "voice": "en-US-JennyNeural"},
     "2": {"name": "Professional Male", "voice": "en-US-GuyNeural"},
@@ -27,25 +30,66 @@ VOICE_STYLES = {
     "5": {"name": "Friendly Female", "voice": "en-US-AnaNeural"},
 }
 
-# CSV path (update with your actual path)
-CSV_PATH = "/home/container/prompts.csv"  # Change this to your server path
+# Store prompts
+csv_prompts = []
 
 # ============================================================
 # BOT SETUP
 # ============================================================
 intents = discord.Intents.default()
 intents.message_content = True
+intents.attachments = True
 bot = commands.Bot(command_prefix="!", intents=intents)
 
 # ============================================================
-# VOICEOVER GENERATION FUNCTION
+# LOAD CSV FROM GENERAL CHANNEL
+# ============================================================
+async def load_csv_from_general():
+    """Read CSV file from #general channel"""
+    global csv_prompts
+    
+    # Find #general channel
+    channel = None
+    for c in bot.get_all_channels():
+        if c.name == CSV_CHANNEL_NAME and isinstance(c, discord.TextChannel):
+            channel = c
+            break
+    
+    if channel is None:
+        return False, f"❌ #{CSV_CHANNEL_NAME} channel not found!"
+    
+    print(f"📁 Searching in #{channel.name}...")
+    
+    # Find message with CSV attachment
+    async for message in channel.history(limit=200):
+        if message.attachments:
+            for attachment in message.attachments:
+                if attachment.filename.endswith('.csv'):
+                    csv_content = await attachment.read()
+                    print(f"📄 Found CSV: {attachment.filename}")
+                    
+                    # Parse CSV
+                    csv_text = csv_content.decode('utf-8')
+                    csv_reader = csv.DictReader(io.StringIO(csv_text))
+                    
+                    csv_prompts = []
+                    for row in csv_reader:
+                        prompt = row.get('prompt', row.get('prompt_text', row.get('text', '')))
+                        if prompt:
+                            csv_prompts.append(prompt)
+                    
+                    return True, f"✅ Loaded {len(csv_prompts)} prompts from #{channel.name}"
+    
+    return False, "❌ No CSV file found in #general channel!"
+
+# ============================================================
+# VOICEOVER GENERATION
 # ============================================================
 def generate_natural_voiceover(raw_prompt):
     """Convert technical prompt to natural voiceover"""
-    import random
     
-    intros = ["Watch how", "See how", "Discover how", "Learn how"]
-    outros = ["That's engineering in action!", "Simple mechanics, powerful results!"]
+    intros = ["Watch how", "See how", "Discover how", "Learn how", "Ever wondered how"]
+    outros = ["That's engineering in action!", "Simple mechanics, powerful results!", "That's how industry gets things done!"]
     
     if " in " in raw_prompt:
         parts = raw_prompt.split(" in ", 1)
@@ -71,138 +115,102 @@ async def text_to_speech(text, voice_id, output_path):
     await communicate.save(output_path)
     return output_path
 
-# ============================================================
-# READ CSV FUNCTION
-# ============================================================
-def read_prompts_from_csv(csv_path):
-    """Read prompts from CSV file"""
-    prompts = []
-    
-    if not os.path.exists(csv_path):
-        return None, f"❌ CSV file not found at: {csv_path}"
-    
-    try:
-        with open(csv_path, 'r', encoding='utf-8') as f:
-            reader = csv.DictReader(f)
-            for row in reader:
-                prompt = row.get('prompt', row.get('prompt_text', row.get('text', '')))
-                if prompt:
-                    prompts.append(prompt)
-        
-        return prompts, f"✅ Found {len(prompts)} prompts in CSV"
-    except Exception as e:
-        return None, f"❌ Error reading CSV: {str(e)}"
-
 # ============================================================
 # DISCORD COMMANDS
 # ============================================================
 
 @bot.event
 async def on_ready():
+    print("=" * 50)
     print(f"✅ Bot is ready!")
     print(f"🤖 Name: {bot.user.name}")
-    print(f"📁 CSV Path: {CSV_PATH}")
+    print(f"📁 Looking in: #{CSV_CHANNEL_NAME} channel")
+    
+    # Load CSV on startup
+    status, msg = await load_csv_from_general()
+    print(msg)
     print("=" * 50)
 
-@bot.command(name="ping")
-async def ping(ctx):
-    """Check if bot is alive"""
-    await ctx.send(f"🏓 Pong! Latency: {round(bot.latency * 1000)}ms")
+@bot.command(name="reload")
+async def reload_csv(ctx):
+    """Reload CSV from #general channel"""
+    status, msg = await load_csv_from_general()
+    await ctx.send(msg)
 
 @bot.command(name="list")
-async def list_prompts(ctx):
-    """List all prompts from CSV"""
-    prompts, msg = read_prompts_from_csv(CSV_PATH)
+async def list_prompts(ctx, page: int = 1):
+    """List prompts from CSV
+    Example: !list 2
+    """
+    if not csv_prompts:
+        await ctx.send("❌ No prompts loaded! Use `!reload` first.")
+        return
+    
+    per_page = 10
+    total_pages = (len(csv_prompts) + per_page - 1) // per_page
     
-    if prompts is None:
-        await ctx.send(msg)
+    if page < 1 or page > total_pages:
+        await ctx.send(f"❌ Page {page} not found. Total pages: {total_pages}")
         return
     
-    # Send first 10 prompts
-    response = f"**📋 Total Prompts:** {len(prompts)}\n\n"
-    for i, p in enumerate(prompts[:10], 1):
-        response += f"`{i}. {p[:80]}...`\n"
+    start = (page - 1) * per_page
+    end = min(start + per_page, len(csv_prompts))
     
-    if len(prompts) > 10:
-        response += f"\n*... and {len(prompts) - 10} more*"
+    response = f"**📋 Prompts (Page {page}/{total_pages})** - Total: {len(csv_prompts)}\n\n"
+    for i in range(start, end):
+        response += f"`{i+1}. {csv_prompts[i][:70]}...`\n"
     
     await ctx.send(response)
 
+@bot.command(name="count")
+async def count_prompts(ctx):
+    """Get total number of prompts"""
+    if not csv_prompts:
+        await ctx.send("❌ No prompts loaded! Use `!reload` first.")
+    else:
+        await ctx.send(f"📊 Total prompts in CSV: **{len(csv_prompts)}**")
+
 @bot.command(name="generate")
-async def generate_prompt(ctx, prompt_id: int = None):
-    """Generate voiceover for a specific prompt ID or all"""
-    
-    prompts, msg = read_prompts_from_csv(CSV_PATH)
+async def generate_prompt(ctx, prompt_id: int):
+    """Generate voiceover for a specific prompt
+    Example: !generate 5
+    """
+    if not csv_prompts:
+        await ctx.send("❌ No prompts loaded! Use `!reload` first.")
+        return
     
-    if prompts is None:
-        await ctx.send(msg)
+    if prompt_id < 1 or prompt_id > len(csv_prompts):
+        await ctx.send(f"❌ Prompt ID must be between 1 and {len(csv_prompts)}")
         return
     
-    if prompt_id is not None:
-        # Generate single prompt
-        if prompt_id < 1 or prompt_id > len(prompts):
-            await ctx.send(f"❌ Prompt ID must be between 1 and {len(prompts)}")
-            return
-        
-        prompt_text = prompts[prompt_id - 1]
-        
-        await ctx.send(f"🎤 Generating voiceover for **Prompt #{prompt_id}**...\n`{prompt_text[:100]}...`")
-        
-        # Generate natural voiceover
-        voiceover_text = generate_natural_voiceover(prompt_text)
-        
-        # Generate audio
-        with tempfile.NamedTemporaryFile(delete=False, suffix=".mp3") as tmp:
-            await text_to_speech(voiceover_text, "en-US-JennyNeural", tmp.name)
-            
-            # Send audio
-            await ctx.send(f"✅ **Prompt #{prompt_id}**\n📝 Script: {voiceover_text[:200]}...")
-            await ctx.send(file=discord.File(tmp.name, filename=f"prompt_{prompt_id}.mp3"))
+    prompt_text = csv_prompts[prompt_id - 1]
     
-    else:
-        # Generate all prompts
-        await ctx.send(f"🎤 Generating voiceovers for **{len(prompts)}** prompts... This may take a while!")
-        
-        success_count = 0
-        for idx, prompt_text in enumerate(prompts, 1):
-            voiceover_text = generate_natural_voiceover(prompt_text)
-            
-            with tempfile.NamedTemporaryFile(delete=False, suffix=".mp3") as tmp:
-                await text_to_speech(voiceover_text, "en-US-JennyNeural", tmp.name)
-                await ctx.send(file=discord.File(tmp.name, filename=f"prompt_{idx}.mp3"))
-            
-            success_count += 1
-            await asyncio.sleep(1)  # Rate limit
-        
-        await ctx.send(f"✅ Completed! Generated {success_count} voiceovers!")
+    await ctx.send(f"🎤 **Generating voiceover for Prompt #{prompt_id}**...\n📝 `{prompt_text[:100]}...`")
+    
+    voiceover_text = generate_natural_voiceover(prompt_text)
+    
+    with tempfile.NamedTemporaryFile(delete=False, suffix=".mp3") as tmp:
+        await text_to_speech(voiceover_text, "en-US-JennyNeural", tmp.name)
+        await ctx.send(file=discord.File(tmp.name, filename=f"prompt_{prompt_id}.mp3"))
 
 @bot.command(name="voice")
 async def generate_with_voice(ctx, prompt_id: int, voice_style: str = "1"):
-    """Generate voiceover with specific voice style
-    Usage: !voice 1 2 (prompt_id=1, voice_style=2)
-    
-    Voice styles:
-    1 = Professional Female
-    2 = Professional Male  
-    3 = British Female
-    4 = Deep Male
-    5 = Friendly Female
+    """Generate with specific voice style
+    Example: !voice 5 2
+    Styles: 1=Female, 2=Male, 3=British, 4=Deep, 5=Friendly
     """
-    
-    prompts, msg = read_prompts_from_csv(CSV_PATH)
-    
-    if prompts is None:
-        await ctx.send(msg)
+    if not csv_prompts:
+        await ctx.send("❌ No prompts loaded! Use `!reload` first.")
         return
     
-    if prompt_id < 1 or prompt_id > len(prompts):
-        await ctx.send(f"❌ Prompt ID must be between 1 and {len(prompts)}")
+    if prompt_id < 1 or prompt_id > len(csv_prompts):
+        await ctx.send(f"❌ Prompt ID must be between 1 and {len(csv_prompts)}")
         return
     
     voice = VOICE_STYLES.get(voice_style, VOICE_STYLES["1"])
-    prompt_text = prompts[prompt_id - 1]
+    prompt_text = csv_prompts[prompt_id - 1]
     
-    await ctx.send(f"🎤 Generating with **{voice['name']}** voice for Prompt #{prompt_id}...")
+    await ctx.send(f"🎤 Generating with **{voice['name']}** for Prompt #{prompt_id}...")
     
     voiceover_text = generate_natural_voiceover(prompt_text)
     
@@ -211,31 +219,28 @@ async def generate_with_voice(ctx, prompt_id: int, voice_style: str = "1"):
         await ctx.send(file=discord.File(tmp.name, filename=f"prompt_{prompt_id}.mp3"))
 
 @bot.command(name="batch")
-async def batch_generate(ctx, start_id: int = 1, end_id: int = None, voice_style: str = "1"):
+async def batch_generate(ctx, start_id: int, end_id: int = None, voice_style: str = "1"):
     """Generate voiceovers for a range of prompts
-    Usage: !batch 1 10 2
+    Example: !batch 1 10
     """
-    
-    prompts, msg = read_prompts_from_csv(CSV_PATH)
-    
-    if prompts is None:
-        await ctx.send(msg)
+    if not csv_prompts:
+        await ctx.send("❌ No prompts loaded! Use `!reload` first.")
         return
     
     if end_id is None:
-        end_id = len(prompts)
+        end_id = len(csv_prompts)
     
-    if start_id < 1 or end_id > len(prompts) or start_id > end_id:
-        await ctx.send(f"❌ Invalid range. Use 1 to {len(prompts)}")
+    if start_id < 1 or end_id > len(csv_prompts) or start_id > end_id:
+        await ctx.send(f"❌ Invalid range. Use 1 to {len(csv_prompts)}")
         return
     
     voice = VOICE_STYLES.get(voice_style, VOICE_STYLES["1"])
     total = end_id - start_id + 1
     
-    await ctx.send(f"🎤 Generating {total} voiceovers (IDs {start_id}-{end_id}) with **{voice['name']}** voice...")
+    await ctx.send(f"🎤 Generating {total} voiceovers (IDs {start_id}-{end_id}) with **{voice['name']}**...")
     
     for idx in range(start_id, end_id + 1):
-        prompt_text = prompts[idx - 1]
+        prompt_text = csv_prompts[idx - 1]
         voiceover_text = generate_natural_voiceover(prompt_text)
         
         with tempfile.NamedTemporaryFile(delete=False, suffix=".mp3") as tmp:
@@ -246,43 +251,60 @@ async def batch_generate(ctx, start_id: int = 1, end_id: int = None, voice_style
     
     await ctx.send(f"✅ Completed! Generated {total} voiceovers!")
 
-@bot.command(name="count")
-async def count_prompts(ctx):
-    """Get total number of prompts in CSV"""
-    prompts, msg = read_prompts_from_csv(CSV_PATH)
+@bot.command(name="random")
+async def random_prompt(ctx):
+    """Generate voiceover for a random prompt"""
+    if not csv_prompts:
+        await ctx.send("❌ No prompts loaded! Use `!reload` first.")
+        return
     
-    if prompts is None:
-        await ctx.send(msg)
-    else:
-        await ctx.send(f"📊 Total prompts in CSV: **{len(prompts)}**")
+    prompt_id = random.randint(1, len(csv_prompts))
+    prompt_text = csv_prompts[prompt_id - 1]
+    
+    await ctx.send(f"🎲 **Random Prompt #{prompt_id}**: `{prompt_text[:100]}...`")
+    
+    voiceover_text = generate_natural_voiceover(prompt_text)
+    
+    with tempfile.NamedTemporaryFile(delete=False, suffix=".mp3") as tmp:
+        await text_to_speech(voiceover_text, "en-US-JennyNeural", tmp.name)
+        await ctx.send(file=discord.File(tmp.name, filename=f"random_{prompt_id}.mp3"))
 
 @bot.command(name="info")
 async def bot_info(ctx):
     """Show bot information"""
     embed = discord.Embed(
         title="🤖 Voiceover Generator Bot",
-        description="Generates voiceovers from CSV prompts",
+        description="Generates voiceovers from CSV in #general",
         color=discord.Color.blue()
     )
-    embed.add_field(name="CSV Path", value=CSV_PATH, inline=False)
-    embed.add_field(name="Voice Styles", value=", ".join([f"{k}={v['name']}" for k, v in VOICE_STYLES.items()]), inline=False)
-    embed.add_field(name="Commands", value="""`!list` - List prompts
+    embed.add_field(name="CSV Channel", value="#general", inline=False)
+    embed.add_field(name="Prompts Loaded", value=len(csv_prompts) if csv_prompts else 0, inline=False)
+    embed.add_field(name="Voice Styles", value="\n".join([f"{k}. {v['name']}" for k, v in VOICE_STYLES.items()]), inline=False)
+    embed.add_field(name="Commands", value="""`!reload` - Reload CSV
+`!list` - List prompts
 `!count` - Count prompts
 `!generate <id>` - Generate voiceover
 `!voice <id> <style>` - Generate with style
 `!batch <start> <end>` - Batch generate
-`!ping` - Check bot""", inline=False)
+`!random` - Random prompt
+`!info` - This menu""", inline=False)
     
     await ctx.send(embed=embed)
 
+@bot.command(name="ping")
+async def ping(ctx):
+    """Check if bot is alive"""
+    await ctx.send(f"🏓 Pong! Latency: {round(bot.latency * 1000)}ms")
+
 # ============================================================
 # RUN BOT
 # ============================================================
 if __name__ == "__main__":
-    print("=" * 60)
+    print("=" * 50)
     print("🤖 Discord Voiceover Bot")
-    print("=" * 60)
-    print(f"📁 CSV Path: {CSV_PATH}")
+    print("=" * 50)
+    print("📁 Reading CSV from #general channel")
     print("🚀 Starting bot...")
+    print("=" * 50)
     
     bot.run(TOKEN)