Update app.py

app.py

@@ -14,11 +14,17 @@ intents.message_content = True
 
 bot = commands.Bot(command_prefix="!", intents=intents)
 
-# Mapping flag emojis to language codes (for now we support English only)
-# We'll use English as target, but you can extend this to use different prompts.
+# Mapping flag emojis to their language names
 flag_to_language = {
-    "🇬🇧": "en",  # English
-    # Add more if needed; for now our prompt forces translation to English.
+    "🇬🇧": "English",
+    "🇫🇷": "French",
+    "🇩🇪": "German",
+    "🇪🇸": "Spanish",
+    "🇮🇹": "Italian",
+    "🇷🇺": "Russian",
+    "🇯🇵": "Japanese",
+    "🇨🇳": "Chinese",
+    # Add more flags and language names as needed
 }
 
 @bot.event
@@ -27,7 +33,7 @@ async def on_ready():
 
 @bot.event
 async def on_raw_reaction_add(payload):
-    # Ignore if the reaction was added by the bot itself
+    # Ignore reactions by the bot itself
     if payload.user_id == bot.user.id:
         return
 
@@ -36,7 +42,9 @@ async def on_raw_reaction_add(payload):
         target_lang = flag_to_language[emoji]
         channel = bot.get_channel(payload.channel_id)
         if channel is None:
+            print("Channel not found")
             return
+
         try:
             # Fetch the full message to access its content
             message = await channel.fetch_message(payload.message_id)
@@ -49,17 +57,23 @@ async def on_raw_reaction_add(payload):
             print("Message content is empty!")
             return
 
-        # Build the prompt for the translation AI.
-        prompt = f'Translate "{original_text}" to English. Write only and only the translated text, no need for explanation or anything.'
+        # Build the translation prompt using the target language
+        prompt = (
+            f'Translate "{original_text}" to {target_lang}. '
+            "Write only and only the translated text, no need for explanation or anything."
+        )
         encoded_prompt = urllib.parse.quote(prompt)
         url = f"https://text.pollinations.ai/{encoded_prompt}?model=gemini"
-        
+        print(f"GET {url}")
+
         try:
-            # Use asyncio.to_thread to run the blocking GET request in a separate thread.
-            response = await asyncio.to_thread(requests.get, url)
+            # Run the blocking GET request in a thread so as not to block the event loop.
+            response = await asyncio.to_thread(requests.get, url, timeout=10)
             response.raise_for_status()
             translated_text = response.text.strip()
-            print(translated_text)
+            print(f"API response: {translated_text}")
+            if not translated_text:
+                translated_text = "[No translation returned]"
         except Exception as e:
             print(f"Error during GET request: {e}")
             translated_text = "[Error translating text]"
@@ -67,9 +81,9 @@ async def on_raw_reaction_add(payload):
         await channel.send(f"Translated to {target_lang}: {translated_text}")
 
 def run_discord_bot():
-    bot.run("MTM1MjI2OTQ4MTQwNTE4NjA3MA.GrdeHW.rYndSNvb9mepFdp_uTK4IOAmKwt31QER6hRgzg")  # Replace with your actual token
+    bot.run("MTM1MjI2OTQ4MTQwNTE4NjA3MA.GrdeHW.rYndSNvb9mepFdp_uTK4IOAmKwt31QER6hRgzg")  # Replace with your bot token
 
-# Run Discord bot in a separate daemon thread
+# Run the Discord bot in a separate daemon thread
 threading.Thread(target=run_discord_bot, daemon=True).start()
 
 # Gradio interface function (simple echo interface)