Update app.py

app.py

@@ -38,7 +38,10 @@ MESSAGES = {
         "history_limit_label": "Number of songs to show",
         "no_history": "📋 No song recognition history yet",
         "empty_history": "📋 Song history is empty",
-        "history_error": "❌ Error loading song history: {error}"
+        "history_error": "❌ Error loading song history: {error}",
+        "chat_title": "### Ask me more about this artist or music in general",
+        "chat_placeholder": "Ask about the artist, song, or anything music related...",
+        "chat_button": "Send"
     },
     "es": {
         "title": "# 🎵 Reconocimiento Musical y Datos Curiosos",
@@ -59,7 +62,10 @@ MESSAGES = {
         "history_limit_label": "Número de canciones a mostrar",
         "no_history": "📋 No hay historial de canciones reconocidas todavía",
         "empty_history": "📋 El historial de canciones está vacío",
-        "history_error": "❌ Error al cargar el historial: {error}"
+        "history_error": "❌ Error al cargar el historial: {error}",
+        "chat_title": "### Pregúntame más sobre este artista o música en general",
+        "chat_placeholder": "Pregunta sobre el artista, la canción o cualquier tema musical...",
+        "chat_button": "Enviar"
     },
     "fr": {
         "title": "# 🎵 Reconnaissance Musicale et Anecdotes",
@@ -80,28 +86,28 @@ MESSAGES = {
         "history_limit_label": "Nombre de chansons à afficher",
         "no_history": "📋 Aucun historique de reconnaissance de chansons pour l'instant",
         "empty_history": "📋 L'historique des chansons est vide",
-        "history_error": "❌ Erreur lors du chargement de l'historique : {error}"
+        "history_error": "❌ Erreur lors du chargement de l'historique : {error}",
+        "chat_title": "### Posez-moi des questions sur cet artiste ou la musique en général",
+        "chat_placeholder": "Posez des questions sur l'artiste, la chanson ou tout sujet musical...",
+        "chat_button": "Envoyer"
     }
 }
 
 @tool
 def recognize_song(audio_path: str) -> dict:
-    """Recognizes a song from an audio file using AudD API
+    """Recognizes a song from an audio file
     Args:
         audio_path: path to the audio file to recognize
     """
     AUDD_API_TOKEN = os.getenv("AUDD_API_TOKEN")
-    print(f"Token usado: {AUDD_API_TOKEN}")
     if not os.path.exists(audio_path):
-        print(f"Audio file not found: {audio_path}")
         return {"error": "The audio file does not exist"}
+    
     try:
         with open(audio_path, 'rb') as file:
             data = {'api_token': AUDD_API_TOKEN, 'return': 'spotify,apple_music'}
             files = {'file': file}
-            print(f"Sending request to AudD API for {audio_path}")
             response = requests.post('https://api.audd.io/', data=data, files=files)
-            print(f"AudD Response: {response.status_code} - {response.text}")
         
         if response.status_code != 200:
             return {"error": f"API Error: {response.status_code}"}
@@ -125,19 +131,24 @@ def recognize_song(audio_path: str) -> dict:
         save_to_history(song_data)
         return song_data
     except Exception as e:
-        print(f"Error processing audio {audio_path}: {str(e)}")
         return {"error": f"Error processing audio: {str(e)}"}
 
 def save_to_history(song_data):
+    """Saves a song to the history"""
     try:
         if os.path.exists(HISTORY_FILE):
             with open(HISTORY_FILE, 'r') as f:
-                history = json.load(f)
+                try:
+                    history = json.load(f)
+                except json.JSONDecodeError:
+                    history = []
         else:
             history = []
+        
         history.insert(0, song_data)
         if len(history) > 50:
             history = history[:50]
+        
         with open(HISTORY_FILE, 'w') as f:
             json.dump(history, f, indent=2)
     except Exception as e:
@@ -145,13 +156,21 @@ def save_to_history(song_data):
 
 @tool
 def view_song_history(limit: int = 10, language: str = "en") -> str:
+    """Shows the history of recognized songs
+    Args:
+        limit: maximum number of songs to display (default 10)
+        language: language code (en, es, fr)
+    """
     lang = LANGUAGES.get(language, "en")
     try:
         if not os.path.exists(HISTORY_FILE):
             return f"📋 {MESSAGES[lang]['no_history']}"
         
         with open(HISTORY_FILE, 'r') as f:
-            history = json.load(f)
+            try:
+                history = json.load(f)
+            except json.JSONDecodeError:
+                return f"📋 Error loading song history."
         
         if not history:
             return f"📋 {MESSAGES[lang]['empty_history']}"
@@ -172,6 +191,12 @@ def view_song_history(limit: int = 10, language: str = "en") -> str:
 
 @tool
 def get_artist_info(artist_name: str, song_title: str, language: str = "en") -> dict:
+    """Gets detailed artist info including anecdotes and song creation facts
+    Args:
+        artist_name: name of the artist
+        song_title: title of the song
+        language: language code (en, es, fr)
+    """
     prompts = {
         "en": f"Provide a detailed overview of the music artist '{artist_name}' including general information about their career, anecdotes, fun facts about the group, and specific details about the creation of their song '{song_title}' if available. Include an image URL if possible. Keep it engaging and positive.",
         "es": f"Proporciona una visión detallada del artista musical '{artist_name}' incluyendo información general sobre su carrera, anécdotas, datos curiosos sobre el grupo y detalles específicos sobre la creación de su canción '{song_title}' si están disponibles. Incluye una URL de imagen si es posible. Mantenlo atractivo y positivo.",
@@ -185,7 +210,6 @@ def get_artist_info(artist_name: str, song_title: str, language: str = "en") ->
     
     try:
         response = model(messages)
-        # Simple parsing to extract image URL if present (assuming model might return something like [Image: URL])
         content = response.content
         image_url = None
         if "[Image:" in content:
@@ -197,6 +221,7 @@ def get_artist_info(artist_name: str, song_title: str, language: str = "en") ->
     except Exception as e:
         return {"text": f"Could not retrieve information about {artist_name}: {str(e)}", "image": None}
 
+# Agent configuration
 final_answer = FinalAnswerTool()
 model = HfApiModel(
     max_tokens=2096,
@@ -235,7 +260,7 @@ with gr.Blocks(title="Music Recognition & Fun Facts", css=".large-text {font-siz
         
         with gr.Row():
             with gr.Column(scale=1):
-                audio_input = gr.Audio(type="filepath", visible=False, source="microphone", streaming=False)
+                audio_input = gr.Audio(type="filepath", visible=False, source="microphone")
                 record_btn = gr.Button(
                     lambda: MESSAGES[selected_language.value]["rec_button"],
                     elem_classes="big-button",
@@ -251,6 +276,19 @@ with gr.Blocks(title="Music Recognition & Fun Facts", css=".large-text {font-siz
         artist_info_text = gr.Markdown("")
         artist_info_image = gr.Image(type="url", label="Artist Image")
 
+        gr.Markdown(lambda: MESSAGES[selected_language.value]["chat_title"])
+        chat_history = gr.Chatbot(label="Music Chat")
+        
+        with gr.Row():
+            chat_input = gr.Textbox(
+                placeholder=lambda: MESSAGES[selected_language.value]["chat_placeholder"],
+                label="Your question"
+            )
+            chat_btn = gr.Button(
+                lambda: MESSAGES[selected_language.value]["chat_button"],
+                variant="primary"
+            )
+
         with gr.Row():
             history_limit = gr.Slider(
                 minimum=5, maximum=50, value=10, step=5,
@@ -267,7 +305,6 @@ with gr.Blocks(title="Music Recognition & Fun Facts", css=".large-text {font-siz
 
         def update_audio_status(audio_path):
             lang = LANGUAGES.get(selected_language.value, "en")
-            print(f"Ruta del archivo de audio: {audio_path}")
             if audio_path:
                 return "ready", MESSAGES[lang]["audio_loaded"]
             return "no_audio", MESSAGES[lang]["no_audio"]
@@ -282,7 +319,6 @@ with gr.Blocks(title="Music Recognition & Fun Facts", css=".large-text {font-siz
             
             result = recognize_song(audio_path)
             if "error" in result:
-                print(f"Error desde AudD: {result['error']}")
                 return None, "", "", MESSAGES[lang]["error"].format(error=result['error']), ""
             
             recognition_msg = f"### 🎵 {result['Song']}\n\n"
@@ -298,6 +334,16 @@ with gr.Blocks(title="Music Recognition & Fun Facts", css=".large-text {font-siz
             
             return result, recognition_msg, artist_info_content, MESSAGES[lang]["recognized"], artist_info['image'] or ""
 
+        def process_chat(query, language_name, song_info):
+            lang = LANGUAGES.get(language_name, "en")
+            if not query or not song_info:
+                return []
+            try:
+                response = chat_with_assistant(query, "", lang)
+                return [(query, response)]
+            except Exception as e:
+                return [(query, f"Sorry, I couldn't process your request: {str(e)}")]
+
         def view_history_process(limit, language_name):
             return view_song_history(int(limit), LANGUAGES.get(language_name, "en"))
 
@@ -309,6 +355,9 @@ with gr.Blocks(title="Music Recognition & Fun Facts", css=".large-text {font-siz
                 gr.update(label=MESSAGES[lang]["language_label"]),
                 MESSAGES[lang]["rec_button"],
                 MESSAGES[lang]["recognize_button"],
+                MESSAGES[lang]["chat_title"],
+                gr.update(placeholder=MESSAGES[lang]["chat_placeholder"]),
+                MESSAGES[lang]["chat_button"],
                 gr.update(label=MESSAGES[lang]["history_limit_label"]),
                 MESSAGES[lang]["history_button"]
             )
@@ -316,7 +365,7 @@ with gr.Blocks(title="Music Recognition & Fun Facts", css=".large-text {font-siz
         language_dropdown.change(
             fn=update_language,
             inputs=[language_dropdown],
-            outputs=[demo.get_component(0), demo.get_component(1), language_dropdown, record_btn, recognize_btn, history_limit, view_history_btn]
+            outputs=[demo.get_component(0), demo.get_component(1), language_dropdown, record_btn, recognize_btn, demo.get_component(8), chat_input, chat_btn, history_limit, view_history_btn]
         )
         
         record_btn.click(
@@ -337,6 +386,16 @@ with gr.Blocks(title="Music Recognition & Fun Facts", css=".large-text {font-siz
             outputs=[song_info_state, recognition_results, artist_info_text, audio_status_msg, artist_info_image]
         )
         
+        chat_btn.click(
+            fn=process_chat,
+            inputs=[chat_input, language_dropdown, song_info_state],
+            outputs=[chat_history]
+        ).then(
+            fn=lambda: "",
+            inputs=[],
+            outputs=[chat_input]
+        )
+        
         view_history_btn.click(
             fn=view_history_process,
             inputs=[history_limit, language_dropdown],