Recent updates (WIP)

Gradio_UI.py

@@ -178,7 +178,7 @@ def stream_to_gradio(
             content={"path": final_answer.to_string(), "mime_type": "audio/wav"},
         )
     else:
-        yield gr.ChatMessage(role="assistant", content=f"**Final answer:** {str(final_answer)}")
+        yield gr.ChatMessage(role="assistant", content=f"**Final answer:** \n{str(final_answer)}")
 
 
 class GradioUI:

app.py

@@ -54,7 +54,7 @@ def main():
     # Determine if we're in test mode (local) or production (HuggingFace)
     # HuggingFace environment has SPACE_ID environment variable
     
-    # Инструкции агенту настраиваются непосредственно в GradioUI.py
+   
     if use_localhost:
         launch_kwargs = {
             "debug": True,
@@ -63,14 +63,12 @@ def main():
             "server_port": 3000
         }
     else:
-        # Configuration for production (HuggingFace)
         launch_kwargs = {
             "debug": True,
-            "share": True
-            # No server_name or server_port for HuggingFace deployment
+            "share": False
         }
     
-    # Передаем инструкцию в конструктор GradioUI
+    # Инструкции агенту настраиваются непосредственно в GradioUI.py
     GradioUI(single_agent).launch(**launch_kwargs)
     logger.success("Server started successfully")
 

config.py

@@ -43,10 +43,13 @@ def get_model_id(provider="gemini"):
     elif provider == "gemini":
         return "gemini/gemini-2.0-flash"
     elif provider == "openrouter":
-        return "openrouter/google/gemini-2.0-flash-lite-preview-02-05:free"  # OpenRouter Claude 3 Opus
+        # return "openrouter/google/gemini-2.0-flash-lite-preview-02-05:free"
         # return "openrouter/mistralai/mistral-small-3.1-24b-instruct:free"
-        #return "openrouter/rekaai/reka-flash-3:free"
         # return "openrouter/deepseek/deepseek-chat:free"
+        # return "openrouter/thudm/glm-z1-32b:free"
+        # return "openrouter/rekaai/reka-flash-3:free"
+        # return "openrouter/google/gemini-2.5-pro-exp-03-25:free"
+        return "openrouter/google/gemini-2.0-flash-exp:free"
 
 def get_ollama_api_base():
     """Get the API base URL for Ollama."""

list_models.py

@@ -0,0 +1,33 @@
+#!/usr/bin/env python3
+"""List available Gemini models"""
+
+import os
+import google.generativeai as genai
+from dotenv import load_dotenv
+
+load_dotenv()
+
+try:
+    # Настройка API ключа
+    api_key = os.environ.get("GEMINI_API_KEY")
+    if not api_key:
+        print("GEMINI_API_KEY not found in environment variables")
+        exit(1)
+        
+    genai.configure(api_key=api_key)
+
+    # Получение списка моделей
+    models = list(genai.list_models())
+    print(f"Found {len(models)} models:")
+    for model in models:
+        print(f"\nModel name: {model.name}")
+        print(f"Display name: {model.display_name}")
+        print(f"Description: {model.description}")
+        print(f"Generation methods: {', '.join(model.supported_generation_methods) if hasattr(model, 'supported_generation_methods') else 'Not specified'}")
+        print(f"Input token limit: {model.input_token_limit}")
+        print(f"Output token limit: {model.output_token_limit}")
+        print(f"Temperature: {model.temperature}") 
+        print("=" * 80)
+        
+except Exception as e:
+    print(f"Error listing models: {e}")

run_single_agent.py

@@ -30,12 +30,12 @@ else:
 # model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud' 
 
 # Prompt the user for the song name
-song_data = "Mechanical Me - Beachy Head"
+song_data = "Ascending - Unna notte che non passa"
 
 agent = create_single_agent(model)
 prompt = f"""1. Find and extract the lyrics of the song: {song_data}.
-2. Perform deep lyrics analysis and return full lyrics and analysis results 
-  in a pretty human-readable format.
+2. Perform deep lyrics analysis and add translation in case it is not in English. 
+3. Return full lyrics and analysis results in a pretty human-readable format.
 """
 # Agent execution
 agent.run(prompt)

test_analysis.py

@@ -0,0 +1,68 @@
+#!/usr/bin/env python3
+"""
+Test script for the lyrics analysis tool.
+This script demonstrates the usage of AnalyzeLyricsTool with sample lyrics.
+"""
+
+from tools.analysis_tools import AnalyzeLyricsTool
+
+# Sample lyrics for testing
+SAMPLE_LYRICS = """
+Walking alone through empty streets
+The silence echoes what my heart repeats
+Shadows dance across the walls
+As night descends and darkness calls
+
+But I'll keep moving forward
+Even when the path is unclear
+There's strength in the journey
+And meaning in the tears
+
+The morning light breaks through the clouds
+Illuminating what was hidden in shrouds
+Each step I take leaves footprints behind
+Marking the path for others to find
+
+But I'll keep moving forward
+Even when the path is unclear
+There's strength in the journey
+And meaning in the tears
+"""
+
+def test_lyrics_analysis():
+    """Test the lyrics analysis tool with sample lyrics."""
+    print("\n=== TESTING LYRICS ANALYSIS TOOL ===\n")
+    
+    # Create an instance of the analysis tool
+    analyzer = AnalyzeLyricsTool()
+    
+    # Analyze the sample lyrics
+    song_title = "The Journey"
+    artist = "Example Artist"
+    
+    # Test with formatting enabled (default)
+    print("\n--- WITH RICH FORMATTING ---\n")
+    formatted_analysis = analyzer.forward(
+        song_title=song_title,
+        artist=artist,
+        lyrics=SAMPLE_LYRICS,
+        format_output=True
+    )
+    print(formatted_analysis)
+    
+    # Test with formatting disabled (raw JSON)
+    print("\n--- WITHOUT FORMATTING (RAW JSON) ---\n")
+    raw_analysis = analyzer.forward(
+        song_title=song_title,
+        artist=artist,
+        lyrics=SAMPLE_LYRICS,
+        format_output=False
+    )
+    print(raw_analysis)
+    
+    return formatted_analysis, raw_analysis
+
+if __name__ == "__main__":
+    print("Testing AnalyzeLyricsTool with sample lyrics...")
+    formatted_result, raw_result = test_lyrics_analysis()
+    print("\nAnalysis test completed successfully!")

test_formatting.py

@@ -0,0 +1,85 @@
+#!/usr/bin/env python3
+"""
+Test script for the formatting tools.
+This script demonstrates the usage of FormatAnalysisResultsTool with sample lyrics analysis.
+"""
+
+from tools.formatting_tools import FormatAnalysisResultsTool
+
+# Sample lyrics analysis in JSON format
+SAMPLE_ANALYSIS = {
+    "summary": "This song explores themes of isolation, introspection, and the search for meaning in a chaotic world. It combines melancholic imagery with moments of hope and resilience.",
+    "main_themes": ["Isolation", "Self-discovery", "Resilience", "Nature as metaphor"],
+    "mood": "Melancholic yet hopeful",
+    "sections_analysis": [
+        {
+            "section_type": "verse",
+            "section_number": 1,
+            "lines": [
+                "Walking alone through empty streets",
+                "The silence echoes what my heart repeats",
+                "Shadows dance across the walls",
+                "As night descends and darkness calls"
+            ],
+            "analysis": "The opening verse establishes the solitary mood through imagery of empty streets and silence. The speaker appears to be in a state of introspection, with external elements (shadows, darkness) reflecting their internal emotional state."
+        },
+        {
+            "section_type": "chorus",
+            "section_number": 1,
+            "lines": [
+                "But I'll keep moving forward",
+                "Even when the path is unclear",
+                "There's strength in the journey",
+                "And meaning in the tears"
+            ],
+            "analysis": "The chorus shifts to a more resilient tone, introducing the central theme of perseverance despite uncertainty. It suggests that the process of struggling itself has value, contrasting with the melancholy of the verse."
+        },
+        {
+            "section_type": "verse",
+            "section_number": 2,
+            "lines": [
+                "The morning light breaks through the clouds",
+                "Illuminating what was hidden in shrouds",
+                "Each step I take leaves footprints behind",
+                "Marking the path for others to find"
+            ],
+            "analysis": "The second verse introduces imagery of dawn and light, symbolizing hope and clarity emerging from darkness. There's a suggestion that the speaker's personal journey might serve as guidance for others, adding a layer of purpose to their struggles."
+        }
+    ],
+    "significant_lines": [
+        {
+            "line": "There's strength in the journey, And meaning in the tears",
+            "significance": "This line encapsulates the central message of finding value in difficult experiences and emotional pain."
+        },
+        {
+            "line": "Each step I take leaves footprints behind, Marking the path for others to find",
+            "significance": "This suggests that personal struggles can have broader meaning by helping others navigate similar challenges."
+        }
+    ],
+    "conclusion": "The song ultimately presents a nuanced view of personal struggle, suggesting that periods of isolation and difficulty are not merely obstacles to overcome but meaningful parts of human experience. Through its imagery of darkness giving way to light, it offers a message of hope without dismissing the reality of pain."
+}
+
+def test_rich_formatting():
+    """Test the rich formatting output."""
+    formatter = FormatAnalysisResultsTool()
+    rich_output = formatter.forward(SAMPLE_ANALYSIS, pretty=True)
+    print("\n=== RICH FORMATTING OUTPUT ===\n")
+    print(rich_output)
+    return rich_output
+
+def test_simple_formatting():
+    """Test the simple text formatting output."""
+    formatter = FormatAnalysisResultsTool()
+    simple_output = formatter.forward(SAMPLE_ANALYSIS, pretty=False)
+    print("\n=== SIMPLE TEXT FORMATTING OUTPUT ===\n")
+    print(simple_output)
+    return simple_output
+
+if __name__ == "__main__":
+    print("Testing FormatAnalysisResultsTool with sample lyrics analysis...")
+    
+    # Test both formatting options
+    rich_result = test_rich_formatting()
+    simple_result = test_simple_formatting()
+    
+    print("\nFormatting test completed successfully!")

tools/analysis_tools.py

@@ -1,12 +1,12 @@
 """
 Analysis tools for understanding and interpreting song lyrics.
 """
-
+import os
 from loguru import logger
 from smolagents import Tool
 import json
 from typing import Dict
-
+from config import get_model_id
 from api_utils import make_api_call_with_retry
 
 
@@ -24,7 +24,7 @@ class AnalyzeLyricsTool(Tool):
     
     def __init__(self, model_id=None):
         super().__init__()
-        self.model_id = model_id or "gemini/gemini-2.0-flash"
+        self.model_id = get_model_id(provider='openrouter')
     
     def forward(self, song_title: str, artist: str, lyrics: str) -> str:
         """