Update app.py

app.py

@@ -1,85 +1,102 @@
 import gradio as gr
-import lyricsgenius
 import os
 from langdetect import detect
-
-# Get the Genius API token from an environment variable
-GENIUS_API_TOKEN = os.environ.get("GENIUS_API_TOKEN")
-
-# Initialize the lyricsgenius client
-genius = lyricsgenius.Genius(GENIUS_API_TOKEN, verbose=False, timeout=15)
-
-def get_song_info(song_name: str, artist_name: str):
+from transformers import pipeline
+from pypinyin import pinyin, Style
+from janome.tokenizer import Tokenizer
+import re
+
+# We'll use a placeholder for Youtube for now.
+# We will implement this in the next step.
+def search_youtube_video(query):
+    # Placeholder for the next step
+    return f'<iframe width="560" height="315" src="https://www.youtube.com/embed/dQw4w9WgXcQ" frameborder="0" allowfullscreen></iframe>'
+
+# Hugging Face translation models
+# We'll use a simple, general-purpose multilingual model for now.
+# This might not be perfect, but it's a great starting point.
+translation_pipeline = pipeline("translation", model="Helsinki-NLP/opus-mt-mul-en")
+
+# Pinyin Romanization for Chinese
+def get_pinyin_from_chinese(lyrics):
+    """Converts Chinese characters to Pinyin with tone numbers."""
+    pinyin_result = pinyin(lyrics, style=Style.TONE)
+    # Flatten the list of lists and join the pinyin with spaces.
+    return " ".join(["".join(word) for word in pinyin_result])
+
+# Romanization for Japanese (Romaji)
+def get_romaji_from_japanese(lyrics):
+    """Converts Japanese to Romaji using a tokenizer."""
+    t = Tokenizer()
+    romaji_lyrics = []
+    # Tokenize the lyrics and convert each word to its romaji form.
+    for token in t.tokenize(lyrics):
+        romaji_lyrics.append(token.reading.lower().replace("*", ""))
+    return " ".join(romaji_lyrics)
+
+# The main function to handle all the logic
+def process_lyrics(lyrics: str, song_name: str = "", artist_name: str = ""):
     """
-    Searches for a song's lyrics, determines the language, and gets a YouTube link.
-
-    Args:
-        song_name: The name of the song to search for.
-        artist_name: The name of the artist (optional).
-
-    Returns:
-        A tuple containing the song's original lyrics, a placeholder translation,
-        and the YouTube embed HTML.
+    Takes user-pasted lyrics and performs language detection, translation,
+    and Romanization if necessary.
     """
-    if not song_name:
-        return "Please enter a song name.", "", ""
-
-    # Search for the song using lyricsgenius
-    try:
-        # --- MODIFIED: Pass both song_name and artist_name to the search method ---
-        song = genius.search_song(song_name, artist_name, get_full_info=False)
-        if not song:
-            return "Song not found.", "", ""
-    except Exception as e:
-        return f"Error searching for song: {e}", "", ""
-
-    original_lyrics = song.lyrics
+    if not lyrics:
+        return "Please paste some lyrics.", "", ""
 
     # Detect the language of the lyrics
     try:
-        language = detect(original_lyrics)
+        language = detect(lyrics)
     except Exception:
         language = "unknown"
 
-    # Placeholder for translation/romanization
-    translated_lyrics = ""
-    if language == "zh":  # Chinese
-        translated_lyrics = "Pinyin placeholder and a translation here."
-    elif language == "ja":  # Japanese
-        translated_lyrics = "Romanji placeholder and a translation here."
-    elif language == "en":  # English
-        translated_lyrics = "Original lyrics are in English."
+    translated_text = ""
+    # We will use the translation model only if the language is not English.
+    if language != "en":
+        try:
+            # We must specify the source language for the model to work properly.
+            # `translation_pipeline` works with language codes like `>>zh<<`.
+            source_language_tag = f">>{language}<<"
+            translation_result = translation_pipeline(lyrics, src_lang=language, tgt_lang="en")[0]['translation_text']
+            translated_text = translation_result
+        except Exception as e:
+            translated_text = f"Translation error: {e}"
     else:
-        translated_lyrics = f"Translation for language: {language}"
+        translated_text = "Lyrics are in English."
+
+    # Handle Romanization based on the detected language
+    if language == "zh":
+        romanized_text = get_pinyin_from_chinese(lyrics)
+        translated_text = f"Pinyin: {romanized_text}\n\nTranslation: {translated_text}"
+    elif language == "ja":
+        romanized_text = get_romaji_from_japanese(lyrics)
+        translated_text = f"Romaji: {romanized_text}\n\nTranslation: {translated_text}"
 
-    # Placeholder for YouTube embed
-    # In a real-world scenario, you would use a library to search for the song on YouTube.
-    youtube_url = "https://www.youtube.com/embed/dQw4w9WgXcQ" # A classic Rick Roll for now!
-    youtube_embed = f'<iframe width="560" height="315" src="{youtube_url}" frameborder="0" allowfullscreen></iframe>'
+    # Get the YouTube video embed (we'll make this real in the next step)
+    youtube_embed = search_youtube_video(f"{song_name} {artist_name} lyrics")
 
-    return original_lyrics, translated_lyrics, youtube_embed
+    return lyrics, translated_text, youtube_embed
 
 # Define the Gradio interface
 with gr.Blocks() as demo:
     gr.Markdown("## Song Lyrics and Translation App")
-    gr.Markdown("Search for a song name to get its lyrics, translation, and a YouTube video.")
+    gr.Markdown("Paste lyrics below and get a translation, romanization, and a YouTube video.")
 
     with gr.Row():
-        song_input = gr.Textbox(label="Enter Song Name", placeholder="e.g., Bohemian Rhapsody")
-        # --- NEW: Add a textbox for the artist name ---
-        artist_input = gr.Textbox(label="Enter Artist Name (Optional)", placeholder="e.g., Queen")
-        search_button = gr.Button("Search")
-
-    with gr.Column():
-        lyrics_output = gr.Textbox(label="Original Lyrics", lines=10)
-        translation_output = gr.Textbox(label="Translation / Romanization", lines=5)
-        video_embed_output = gr.HTML(label="YouTube Video")
-
-    search_button.click(
-        fn=get_song_info,
-        # --- MODIFIED: Pass both input components to the function ---
-        inputs=[song_input, artist_input],
-        outputs=[lyrics_output, translation_output, video_embed_output]
+        with gr.Column(scale=2):
+            lyrics_input = gr.Textbox(label="Paste Lyrics Here", lines=10, placeholder="Paste your lyrics here...")
+            # We keep the song/artist inputs for the Youtube
+            song_input = gr.Textbox(label="Enter Song Name (for Youtube)", placeholder="e.g., 月亮代表我的心")
+            artist_input = gr.Textbox(label="Enter Artist Name (Optional)", placeholder="e.g., Teresa Teng")
+            process_button = gr.Button("Process")
+        
+        with gr.Column(scale=1):
+            translation_output = gr.Textbox(label="Translation & Romanization", lines=10)
+            video_embed_output = gr.HTML(label="YouTube Video")
+
+    process_button.click(
+        fn=process_lyrics,
+        inputs=[lyrics_input, song_input, artist_input],
+        outputs=[lyrics_input, translation_output, video_embed_output]
     )
 
 # Launch the Gradio app