Upload folder using huggingface_hub

app.py

@@ -1,4 +1,5 @@
 import os
+import time
 import gradio as gr
 from glm_huggingface import run_glm
 from open_ai import run_openai
@@ -26,28 +27,70 @@ SYSPROMT_GEN_LYRICS = get_prompt("SYSPROMT_GEN_LYRICS")
 SYSPROMT_GEN_STYLE_FROM_LYRICS = get_prompt("SYSPROMT_GEN_STYLE_FROM_LYRICS")
 PROMT_GEN_STYLE_FROM_LYRICS = get_prompt("PROMT_GEN_STYLE_FROM_LYRICS")
 
-def get_style_lyrics(style: str, lyrics: str, gen_lyrics: bool, gen_style_from_lyrics: bool = False):
+def get_style_lyrics(style: str, lyrics: str, gen_lyrics: bool, gen_style_from_lyrics: bool):
+    # Detect current UI language from global TS variable
+    current_lang = "RU" if TS == russian_strings else "EN"
+    
+    # Initialize timing and logging variables
+    start_time = time.time()
+    stage_times = {}
+    errors = []
 
+    
+    # Helper function to get localized message
+    def get_msg(key):
+        return TS.get(key, english_strings.get(key, f"Missing key: {key}"))
+    
+    # Helper function to yield progress with timing
+    def yield_progress(message_key, status="🔄", **kwargs):
+        message = get_msg(message_key).format(**kwargs)
+        elapsed = time.time() - start_time
+        progress_msg = f"{status} {message} (⏱️ {elapsed:.2f}s)"
+        print(f"[PROGRESS] {message}")
+        return progress_msg
+
+    log = ''
+    yield ("", "", "🚀" + get_msg("log_starting"))
+    
+    gen_start = time.time()    
+    # Stage 1: Lyrics Generation (if requested)
     if gen_lyrics:
+        stage_start = time.time()
+        yield ("", lyrics, log + " ⏳" + get_msg("log_lyrics_gen_start"))
+
         try:
-            gen_lyrics = run_llm(f"{PROMT_GEN_LYRICS}\n<THEME>{lyrics}</THEME>", sys_prompt=SYSPROMT_GEN_LYRICS)
-            print (gen_lyrics)
-            if gen_lyrics:
-                lyrics=gen_lyrics
+            gen_lyrics_result = run_llm(f"{PROMT_GEN_LYRICS}\n<THEME>{lyrics}</THEME>", sys_prompt=SYSPROMT_GEN_LYRICS)
+            if gen_lyrics_result:
+                lyrics = gen_lyrics_result
             else:
-                lyrics=''
+                lyrics = ''            
         except Exception as e:
-            print(f"ERROR infer llm: {e}")
-            lyrics=''
+            stage_time = time.time() - stage_start
+            stage_times['lyrics_generation'] = stage_time
+            error_msg = f"ERROR infer llm: {e}"
+            errors.append(error_msg)
+            yield yield_progress("log_lyrics_gen_error", status="❌", error=str(e))
+            print(error_msg)
+            lyrics = ''
+        stage_time = time.time() - stage_start
+        stage_times['lyrics_generation'] = stage_time
+        log += " "+yield_progress("log_lyrics_gen_complete", status=" 🖌️", time=stage_time)
+        yield ("", lyrics, log + " ⏳" + get_msg("log_style_gen_start" if not gen_style_from_lyrics else "log_style_desc_start"))
+
 
+    # Stage 2: Style Generation
+    stage_start = time.time()
     if gen_style_from_lyrics:
         try:
-            style_gen = run_llm(f"{PROMT_GEN_STYLE_FROM_LYRICS}\n<SONG_TEXT><LYRICS>\n{lyrics}\n</LYRICS></SONG_TEXT>",
+            style_gen = run_llm(f"{PROMT_GEN_STYLE_FROM_LYRICS}\n<MUSIC_STYLE>{style}</MISIC_STYLE><SONG_TEXT><LYRICS>\n{lyrics}\n</LYRICS></SONG_TEXT>",
                                 sys_prompt=SYSPROMT_GEN_STYLE_FROM_LYRICS)
             if not style_gen:
                 style_gen = ''
         except Exception as e:
-            print(f"ERROR infer llm: {e}")
+            error_msg = f"ERROR infer llm: {e}"
+            errors.append(error_msg)
+            yield yield_progress("log_style_gen_error", status="❌", error=str(e))
+            print(error_msg)
             style_gen = ''
     else:
         try:
@@ -55,10 +98,19 @@ def get_style_lyrics(style: str, lyrics: str, gen_lyrics: bool, gen_style_from_l
             if not style_gen:
                 style_gen = ''
         except Exception as e:
-            print(f"ERROR infer llm: {e}")
+            error_msg = f"ERROR infer llm: {e}"
+            errors.append(error_msg)
+            yield yield_progress("log_style_gen_error", status="❌", error=str(e))
+            print(error_msg)
             style_gen = ''
+    stage_time = time.time() - stage_start
+    stage_times['style_generation'] = stage_time
+    log += " "+ yield_progress("log_style_gen_complete", status=" 🎨", time=stage_time)
+    yield (style_gen, lyrics, log + " ⏳" + get_msg("log_text_format_start"))
 
 
+    # Stage 3: Text Formatting
+    stage_start = time.time()
     try:
         text_glm = run_llm(f"{PROMPT_TEXT}\n{style_gen}\n#Text song:\n{lyrics}\n{PROMPT_TEXT2}", sys_prompt=SYSPROMPT_TEXT)
         if text_glm:
@@ -66,10 +118,21 @@ def get_style_lyrics(style: str, lyrics: str, gen_lyrics: bool, gen_style_from_l
         else:
             text = ''
     except Exception as e:
-        print(f"ERROR infer llm: {e}")
+        stage_time = time.time() - stage_start
+        stage_times['text_formatting'] = stage_time
+        error_msg = f"ERROR infer llm: {e}"
+        errors.append(error_msg)
+        yield yield_progress("log_text_format_error", status="❌", error=str(e))
+        print(error_msg)
         text = ''
 
-    return style_gen, text
+    stage_time = time.time() - stage_start
+    # Final summary
+    total_time = time.time() - gen_start
+    log += " " + yield_progress("log_text_format_complete", status=" 🧾", time=stage_time)
+    yield (style_gen, text, log)
+   
+
 
 isGenarateLyrics=False
 
@@ -88,7 +151,26 @@ english_strings = {
     "style_label": "AI-Generated Style:",
     "song_result_label": "Formated song text:",
     "isGenerateLyrics": "generate lyrics (experemental)",
-    "isGenerateStyle": "generate music style from lyrics"
+    "isGenerateStyle": "generate music style from lyrics",
+    # Progress logging messages
+    "log_starting": "Starting SUNO prompt generation...",
+    "log_lyrics_gen_start": "Generating lyrics from idea...",
+    "log_lyrics_gen_complete": "Lyrics generation completed in {time:.2f}s",
+    "log_lyrics_gen_error": "Error generating lyrics: {error}",
+    "log_style_gen_start": "Generating music style from lyrics...",
+    "log_style_desc_start": "Generating music style description...",
+    "log_style_gen_complete": "Style generation completed in {time:.2f}s",
+    "log_style_gen_error": "Error generating style: {error}",
+    "log_text_format_start": "Formatting song text...",
+    "log_text_format_complete": "Text formatting completed in {time:.2f}s",
+    "log_text_format_error": "Error formatting text: {error}",
+    "log_summary_success": "✅ Generation completed successfully!",
+    "log_summary_error": "❌ Generation completed with errors",
+    "log_summary_total": "Total execution time: {time:.2f}s",
+    "log_summary_stages": "Stage timings: {stages}",
+    "generation_warning_msg": """**⚠️ Note:** the generation model (*GLM*) is not among the strongest ones and was chosen for its availability on *Hugging Face*.  
+If the output looks off, simply try generating again — it usually fixes the issue.
+"""
 }
 
 russian_strings = {
@@ -105,7 +187,26 @@ russian_strings = {
     "style_label": "AI-сгенерированный музыкальный стиль: ",
     "song_result_label": "Отформатированный текст песни: ",
     "isGenerateLyrics": "придумать текст песни (экспериментальная)",
-    "isGenerateStyle": "придумать стиль музыки по тексту песни"
+    "isGenerateStyle": "придумать стиль музыки по тексту песни",
+    # Progress logging messages
+    "log_starting": "Начинаем генерацию SUNO промптов...",
+    "log_lyrics_gen_start": "Генерируем текст песни из идеи...",
+    "log_lyrics_gen_complete": "Генерация текста завершена за {time:.2f}с",
+    "log_lyrics_gen_error": "Ошибка генерации текста: {error}",
+    "log_style_gen_start": "Генерируем музыкальный стиль из текста...",
+    "log_style_desc_start": "Генерируем описание музыкального стиля...",
+    "log_style_gen_complete": "Генерация стиля завершена за {time:.2f}с",
+    "log_style_gen_error": "Ошибка генерации стиля: {error}",
+    "log_text_format_start": "Форматируем текст песни...",
+    "log_text_format_complete": "Форматирование текста завершено за {time:.2f}с",
+    "log_text_format_error": "Ошибка форматирования текста: {error}",
+    "log_summary_success": "✅ Генерация успешно завершена!",
+    "log_summary_error": "❌ Генерация завершена с ошибками",
+    "log_summary_total": "Общее время выполнения: {time:.2f}с",
+    "log_summary_stages": "Время этапов: {stages}",
+    "generation_warning_msg": """
+**⚠️ Примечание:** используется не самая мощная модель генерации (*GLM*), выбранная из-за доступности на *Hugging Face*.  
+Иногда результат может содержать ошибки — чаще всего помогает просто повторная генерация."""    
 }
 
 lang = "EN"
@@ -116,6 +217,7 @@ else:
     TS=russian_strings
 
 def process_lang(selected_lang, isGenerateLyrics, isGenerateStyle):
+    global TS
     lang=selected_lang
     if selected_lang == "RU":
         TS=russian_strings
@@ -134,7 +236,8 @@ def process_lang(selected_lang, isGenerateLyrics, isGenerateStyle):
             gr.update(label=TS["song_result_label"]),
             gr.update(value=TS["generate_button"]),
             gr.update(label=TS["isGenerateLyrics"]),
-            gr.update(label=TS["isGenerateStyle"])
+            gr.update(label=TS["isGenerateStyle"]),
+            gr.update(value=TS["generation_warning_msg"])
         ]
     return message, *ret
 
@@ -180,28 +283,29 @@ with gr.Blocks(title="Suno PromptGen", fill_height=True,
             song_output = gr.Textbox(label=TS["song_result_label"], lines=25, max_lines=25,show_copy_button=True)
 
     with gr.Row(variant="default"):
-            log_text = gr.Textbox(value=status, container=False, lines=1, max_lines=1)
+            log_text = gr.Textbox(value=status, container=False, lines=1, max_lines=3, label="log")
+    warning_msg = gr.Markdown(value=TS["generation_warning_msg"])
 
     generate_button.click(
         fn=get_style_lyrics,
         inputs=[name_input, text_input, isGenerateLyrics, isGenerateStyle],
-        outputs=[style_output, song_output],
+        outputs=[style_output, song_output, log_text],
         api_name="GenSuno",
         show_api=True
-    )   
+    )
 
     radio_lang.change(process_lang, inputs=[radio_lang, isGenerateLyrics, isGenerateStyle],
                 outputs=[log_text,name_input,text_input,style_output,song_output,generate_button,
-                        isGenerateLyrics, isGenerateStyle],
+                        isGenerateLyrics, isGenerateStyle, warning_msg],
                 api_name=False, show_api=False)
 
     isGenerateLyrics.change(process_lang, inputs=[radio_lang, isGenerateLyrics, isGenerateStyle],
                 outputs=[log_text,name_input,text_input,style_output,song_output,generate_button,
-                        isGenerateLyrics, isGenerateStyle],
+                        isGenerateLyrics, isGenerateStyle, warning_msg],
                 api_name=False, show_api=False)
     isGenerateStyle.change(process_lang, inputs=[radio_lang, isGenerateLyrics, isGenerateStyle],
                 outputs=[log_text,name_input,text_input,style_output,song_output,generate_button,
-                        isGenerateLyrics, isGenerateStyle],
+                        isGenerateLyrics, isGenerateStyle, warning_msg],
                 api_name=False, show_api=False)
     
 demo.css = """