Update app.py

app.py

@@ -13,6 +13,10 @@ from huggingface_hub import hf_hub_download
 os.environ['CUDA_VISIBLE_DEVICES'] = '-1'
 os.environ['TF_CPP_MIN_LOG_LEVEL'] = '3'
 
+max_results = 100
+max_output = 50
+
+
 # Cacher le nom du repo
 python_path = hf_hub_download(repo_id=os.environ['REPO_ID'], repo_type="space", filename=os.environ['MODEL_FILE'],
                                 use_auth_token=os.environ['TOKEN'])
@@ -46,18 +50,35 @@ def process(prompt, lang):
         #for i in range(len(I[0])):
         #    print(audio_names[I[0][i]], " with distance ", D[0][i])
         #    print("    url : ", get_url_myma(I[0][i], audio_names, url_dict))
-            
-        return [audio_names[I[0][0]].split('.')[0], get_url_myma(I[0][0], audio_names, url_dict), 
+
+        formated = []
+        output_csv = f"{file_name}_results_text.csv"
+        with open(output_csv, "w") as w:
+            writer = csv.writer(w)
+            count = 0
+            for top in tops:
+                if count > max_output:
+                    break
+        
+                formated.append(file)
+                formated.append(top)
+                #writer.writerow(dict_catalog[file].values())
+                count += 1
+
+        
+        return [output_csv,
+                audio_names[I[0][0]].split('.')[0], get_url_myma(I[0][0], audio_names, url_dict), 
                 audio_names[I[0][1]].split('.')[0], get_url_myma(I[0][1], audio_names, url_dict), 
                 audio_names[I[0][2]].split('.')[0], get_url_myma(I[0][2], audio_names, url_dict), 
                 audio_names[I[0][3]].split('.')[0], get_url_myma(I[0][3], audio_names, url_dict), 
                 audio_names[I[0][4]].split('.')[0], get_url_myma(I[0][4], audio_names, url_dict)]
     except:
-        return ["Error input - please try again", "",
-               "Error input - please try again", "",
-               "Error input - please try again", "",
-               "Error input - please try again", "",
-               "Error input - please try again", "",]
+        return ["",
+                "Error input - please try again", "",
+                "Error input - please try again", "",
+                "Error input - please try again", "",
+                "Error input - please try again", "",
+                "Error input - please try again", "",]
     
 inputs = [gr.Textbox(label="Input", value="type your description", max_lines=2), 
             gr.Radio(label="Language", choices=["en"], value="en")]
@@ -83,7 +104,8 @@ poc_examples = [
           ["90 eurodance beat","en"],
 ]
 
-outputs = [gr.Textbox(label="Track name 1"), gr.Audio(label="Track 1", show_label=False), 
+outputs = [gr.File(),
+           gr.Textbox(label="Track name 1"), gr.Audio(label="Track 1", show_label=False), 
            gr.Textbox(label="Track name 2"), gr.Audio(label="Track 2", show_label=False), 
            gr.Textbox(label="Track name 3"), gr.Audio(label="Track 3", show_label=False), 
            gr.Textbox(label="Track name 4"), gr.Audio(label="Track 4", show_label=False),