Update app.py

app.py

@@ -6,6 +6,8 @@ import os
 import json
 import gradio as gr
 from tools.final_answer import FinalAnswerTool
+import tempfile
+import shutil
 
 # File to save the history of recognized songs
 HISTORY_FILE = "song_history.json"
@@ -373,7 +375,7 @@ with gr.Blocks(title="Music Recognition & Fun Facts", css=".large-text {font-siz
         # Artist info section
         with gr.Row():
             artist_info_text = gr.Markdown("")
-            artist_info_image = gr.Image(type="url")
+            artist_info_image = gr.Image(type="filepath")  # Changed from type="url" to type="filepath"
 
         gr.Markdown(value=lambda: MESSAGES[selected_language.value]["chat_title"])
         chat_history = gr.Chatbot(label="Music Chat")
@@ -399,6 +401,22 @@ with gr.Blocks(title="Music Recognition & Fun Facts", css=".large-text {font-siz
             else:
                 return "no_audio", MESSAGES[lang]["no_audio"]
         
+        # Function to download image from URL and return a local filepath
+        def download_image(url):
+            if not url:
+                return None
+            try:
+                response = requests.get(url, stream=True)
+                response.raise_for_status()
+                # Create a temporary file
+                temp_file = tempfile.NamedTemporaryFile(delete=False, suffix='.jpg')
+                shutil.copyfileobj(response.raw, temp_file)
+                temp_file.close()
+                return temp_file.name
+            except Exception as e:
+                print(f"Error downloading image: {str(e)}")
+                return None
+        
         # Recognition function
         def process_audio(audio_path, language_name, status):
             lang = LANGUAGES.get(language_name, "en")
@@ -450,10 +468,13 @@ with gr.Blocks(title="Music Recognition & Fun Facts", css=".large-text {font-siz
                 # Artist info section title based on language
                 artist_facts_content = f"{MESSAGES[lang]['artist_info_title']}\n\n{artist_info['text']}"
                 
+                # Download the image if URL exists
+                image_filepath = download_image(artist_info['image']) if artist_info['image'] else ""
+                
                 # Update status message
                 status_msg = MESSAGES[lang]["recognized"]
                 
-                return result, recognition_msg, artist_facts_content, status_msg, artist_info['image']
+                return result, recognition_msg, artist_facts_content, status_msg, image_filepath
             
             except Exception as e:
                 return None, "", "", MESSAGES[lang]["error"].format(error=str(e)), ""