Update app.py

app.py

@@ -11,15 +11,18 @@ async def detect(audio):
     recognize_generator = shazam.recognizeSong()
     results = ''
     while True:
-        print(next(recognize_generator))
-        results += str(next(recognize_generator))
-        data_tuple = ast.literal_eval(results)
-        track_info = data_tuple[1]
-        title = track_info['track']['title']
-        subtitle = track_info['track']['subtitle']
-        background_image = track_info['track']['images']['background']
-        url = track_info['track']['url']
-        return background_image, title + '\nby ' + subtitle + '\nin' + url
+        try:
+            print(next(recognize_generator))
+            results += str(next(recognize_generator))
+            data_tuple = ast.literal_eval(results)
+            track_info = data_tuple[1]
+            title = track_info['track']['title']
+            subtitle = track_info['track']['subtitle']
+            background_image = track_info['track']['images']['background']
+            url = track_info['track']['url']
+            return title + '\nby ' + subtitle + '\nin ' + url, background_image
+        except:
+            return 'No results found.', None
 
-iface = gr.Interface(fn=detect, inputs=[gr.Audio(type="filepath", label="Audio File")], outputs=[gr.Image(label="Cover"), "text"])
+iface = gr.Interface(fn=detect, inputs=[gr.Audio(type="filepath", label="Audio File")], outputs=[gr.Text(label="Info"), gr.Image(label="Cover")])
 iface.launch()