Update main.py

main.py

@@ -5,6 +5,8 @@ import os
 import asyncio
 import json
 from datetime import datetime
+import edge_tts
+import io
 
 # Add these constants at the top
 API_URL = "https://api-inference.huggingface.co/models/openai/whisper-large-v3-turbo"
@@ -14,11 +16,30 @@ AI_HEADERS = {
     "Accept": "application/json"
 }
 
+# TTS settings
+DEFAULT_VOICE = "en-IN-NeerjaNeural"
+DEFAULT_RATE = "+25%"
+
 # Initialize message history in memory only
 message_history = []
 
+async def text_to_speech(text, voice=DEFAULT_VOICE, rate=DEFAULT_RATE):
+    """Convert text to speech and return the audio data"""
+    try:
+        communicate = edge_tts.Communicate(text, voice, rate=rate)
+        audio_data = bytes()
+        
+        async for chunk in communicate.stream():
+            if chunk["type"] == "audio":
+                audio_data += chunk["data"]
+        
+        return audio_data
+    except Exception as e:
+        print(f"Error in text-to-speech conversion: {str(e)}")
+        return None
+
 # Create a new Client instance with a custom session name and no local storage
-app = Client(
+myaibot = Client(
     "my_bot",
     api_id=API_ID,
     api_hash=API_HASH,
@@ -35,7 +56,7 @@ def save_message_to_history(user_id, username, message_type, content, bot_respon
     print("Message History Update:", json.dumps(message_data, indent=2))
 
 # Add command to print history
-@app.on_message(filters.command("history"))
+@myaibot.on_message(filters.command("history"))
 async def history_command(client, message):
     if not message_history:
         await message.reply_text("No message history available.")
@@ -56,12 +77,12 @@ async def history_command(client, message):
         await message.reply_text(f"Error retrieving history: {str(e)}")
 
 # Command handler for /start
-@app.on_message(filters.command("start"))
+@myaibot.on_message(filters.command("start"))
 async def start_command(client, message):
     await message.reply_text("Hello! I'm your Telegram bot. Nice to meet you!")
 
 # Command handler for /help
-@app.on_message(filters.command("help"))
+@myaibot.on_message(filters.command("help"))
 async def help_command(client, message):
     help_text = """
 Available commands:
@@ -71,7 +92,7 @@ Available commands:
     await message.reply_text(help_text)
 
 # Message handler for regular text messages
-@app.on_message(filters.text & filters.private & ~filters.command(["start", "help", "info"]))
+@myaibot.on_message(filters.text & filters.private & ~filters.command(["start", "help", "info"]))
 async def echo(client, message):
     try:
         thinking_msg = await message.reply_text("🤔 Thinking about your message...")
@@ -91,7 +112,7 @@ async def echo(client, message):
         await message.reply_text(f"Sorry, I couldn't process your message: {str(e)}")
 
 # Handle photo messages
-@app.on_message(filters.photo)
+@myaibot.on_message(filters.photo)
 async def handle_photo(client, message):
     response = "Nice photo!"
     await message.reply_text(response)
@@ -104,7 +125,7 @@ async def handle_photo(client, message):
     )
 
 # Handle sticker messages
-@app.on_message(filters.sticker)
+@myaibot.on_message(filters.sticker)
 async def handle_sticker(client, message):
     response = "Cool sticker!"
     await message.reply_text(response)
@@ -117,7 +138,7 @@ async def handle_sticker(client, message):
     )
 
 # Custom command example
-@app.on_message(filters.command("info"))
+@myaibot.on_message(filters.command("info"))
 async def info_command(client, message):
     user = message.from_user
     info_text = f"""
@@ -133,7 +154,7 @@ async def transcribe_audio(file_path):
     try:
         with open(file_path, "rb") as f:
             data = f.read()
-        response = requests.post(API_URL, data=data)
+        response = requests.post(API_URL,data=data)
         return response.json().get('text', 'Could not transcribe audio')
     except Exception as e:
         print(f"Error in transcription: {e}")
@@ -207,37 +228,39 @@ async def get_ai_response(text):
         print(f"Error getting AI response: {str(e)}\nFull error: {repr(e)}")
         return "Sorry, I couldn't process your message."
 
-# Update the voice message handler with retry logic
-@app.on_message(filters.voice | filters.audio)
+# Update the voice message handler with TTS response
+@myaibot.on_message(filters.voice | filters.audio)
 async def handle_voice(client, message):
     try:
-        # Send a processing message
         processing_msg = await message.reply_text("🎵 Processing your voice message...")
         
-        # Download the voice message with retry logic
-        max_retries = 3
-        for attempt in range(max_retries):
-            try:
-                message = await app.get_messages(
-                    message.chat.id,
-                    message.id
-                )
-                voice_file = await message.download()
-                break
-            except Exception as e:
-                if attempt == max_retries - 1:
-                    raise e
-                await asyncio.sleep(1)
-        
+        # Download and process voice message
+        voice_file = await message.download()
         transcription = await transcribe_audio(voice_file)
         await message.reply_text(f"🗣️ Transcription:\n\n{transcription}")
         
+        # Get AI response
         thinking_msg = await message.reply_text("🤔 Thinking about your message...")
         ai_response = await get_ai_response(transcription)
         await thinking_msg.delete()
         await message.reply_text(ai_response)
         
-        # Save voice message to history
+        # Convert AI response to speech
+        processing_tts = await message.reply_text("🔊 Converting response to speech...")
+        audio_data = await text_to_speech(ai_response)
+        
+        if audio_data:
+            # Create an in-memory file-like object
+            audio_file = io.BytesIO(audio_data)
+            audio_file.name = "response.mp3"
+            
+            # Send audio directly from memory
+            await message.reply_voice(
+                audio_file,
+                caption="🎵 Voice response"
+            )
+        
+        # Save to history
         save_message_to_history(
             message.from_user.id,
             message.from_user.username,
@@ -248,8 +271,9 @@ async def handle_voice(client, message):
         
         # Clean up
         try:
-            os.remove(voice_file)
+            os.remove(voice_file)  # Still need to remove the downloaded voice file
             await processing_msg.delete()
+            await processing_tts.delete()
         except:
             pass
         
@@ -261,4 +285,4 @@ async def handle_voice(client, message):
 # Run the bot
 if __name__ == "__main__":
     print("Bot is running...")
-    app.run() 
+    myaibot.run()