Update app.py

app.py

@@ -1,60 +1,84 @@
-import gradio as gr
-import subprocess
-import os
-import shutil
-import uuid
-from transformers import pipeline
-from gtts import gTTS
+# We import the Translator class to handle translations and the LANGUAGES dictionary
+# to get the full name of a language from its code (e.g., 'en' -> 'english').
+from googletrans import Translator, LANGUAGES
 
+def detailed_translate(text_to_translate: str, dest_lang: str):
+    """
+    Translates a piece of text and prints a detailed report of the process.
+
+    Args:
+        text_to_translate (str): The text you want to translate.
+        dest_lang (str): The language code to translate into (e.g., 'es', 'fr').
+    """
+    if not text_to_translate.strip():
+        print("Error: Input text cannot be empty.")
+        return
+
+    print("\n🔄  Processing your translation request...")
 
-def translate_video(file_path):
-    
     try:
-        audio_path = os.path.join(file_path, "audio.wav")
+        # 1. Create a Translator object
+        translator = Translator()
+
+        # 2. Perform the translation. The translate() method returns a result object
+        #    that contains all the details we need.
+        translation_result = translator.translate(text_to_translate, dest=dest_lang)
+
+        # 3. Extract details from the result object
+        original_text = translation_result.origin
+        translated_text = translation_result.text
+        detected_lang_code = translation_result.src
+        target_lang_code = translation_result.dest
+
+        # 4. Get the full, human-readable language names from the codes
+        #    We use .get() with a default value in case the code isn't found
+        detected_lang_name = LANGUAGES.get(detected_lang_code, "Unknown").title()
+        target_lang_name = LANGUAGES.get(target_lang_code, "Unknown").title()
+
+
+        # 5. Print the detailed report
+        print("\n" + "="*15 + " Translation Details " + "="*15)
 
-        if not os.path.exists(audio_path):
-             raise FileNotFoundError("Audio extraction failed. yt-dlp did not produce a .wav file.")
+        print(f"\n📄 Original Transcript:")
+        print(f"   \"{original_text}\"")
 
-        # 3. Translate the audio using the whisper-tiny model
-        translator = pipeline(
-            "automatic-speech-recognition",
-            model="openai/whisper-tiny",
-            device="cpu"
-        )
-        
-        translation = translator(audio_path, return_timestamps=True, generate_kwargs={"task": "translate"})
-        
-        translated_text = translation["text"]
+        print(f"\n🔍 Detected Language:")
+        print(f"   {detected_lang_name} (code: {detected_lang_code})")
 
-        if not translated_text:
-            return "No speech was detected in the video.", None, video_path
+        print(f"\n🎯 Target Language:")
+        print(f"   {target_lang_name} (code: {target_lang_code})")
 
-        # 4. Convert translated text to speech using gTTS
-        tts = gTTS(translated_text.strip(), lang='en')
-        translated_audio_path = os.path.join(temp_dir, "translated_audio.mp3")
-        tts.save(translated_audio_path)
+        print(f"\n✨ Translated Text:")
+        print(f"   \"{translated_text}\"")
 
-        return translated_text, translated_audio_path, video_path
+        print("\n" + "="*49)
 
     except Exception as e:
-        gr.Warning(f"An unexpected error occurred: {str(e)}")
-        return f"An error occurred: {str(e)}", None, None
-
-# Create the Gradio interface
-iface = gr.Interface(
-    fn=translate_video,
-    inputs=gr.Video(label="Upload your video to translate"),
-    outputs=[
-        gr.Textbox(label="Translated Text", interactive=False),
-        gr.Audio(label="Translated Audio"),
-        gr.Video(label="Original Video"),
-    ],
-    title="Twitter/X Video Translator",
-    description="Enter a link to a Twitter/X video to translate its audio to English. Handles videos longer than 30 seconds.",
-    allow_flagging="never",
-)
+        print(f"\n❌ An error occurred during translation.")
+        print(f"   Error details: {e}")
+        print("   Please check your internet connection and ensure the language code is valid.")
 
+
+# --- Main part of the script that runs when you execute it ---
 if __name__ == "__main__":
-    if not os.path.exists("downloads"):
-        os.makedirs("downloads")
-    iface.launch()
+    print("--- Welcome to the Detailed Translator! ---")
+    print("You can type 'quit' or 'exit' at any time to close the program.")
+
+    while True:
+        # Get input from the user
+        text_input = input("\nEnter the text you want to translate: ")
+
+        # Check if the user wants to exit
+        if text_input.lower() in ['quit', 'exit']:
+            print("👋 Goodbye!")
+            break
+
+        target_lang_input = input("Enter the target language (e.g., 'es' for Spanish, 'fr' for French, 'ja' for Japanese): ").lower()
+
+        # Check if the user wants to exit
+        if target_lang_input.lower() in ['quit', 'exit']:
+            print("👋 Goodbye!")
+            break
+
+        # Call our detailed translation function
+        detailed_translate(text_input, target_lang_input)