Update app.py

app.py

@@ -56,6 +56,8 @@ def process(prompt, lang):
 
     formated = []
     output_csv = f"{prompt}_results_text.csv"
+    out = [output_csv]
+    
     with open(output_csv, "w") as w:
         writer = csv.writer(w)
         count = 0
@@ -65,16 +67,15 @@ def process(prompt, lang):
     
             #formated.append(get_url_myma(top, audio_names, url_dict)
             #formated.append(audio_names[top].split('.')[0])
-            #writer.writerow(dict_catalog[file].values())
+            writer.writerow(dict_catalog[file].values())
             count += 1
 
-    out = [output_csv]
-    for i in range(max_output):
-        out.append(audio_names[I[0][i]].split('.')[0])
-        out.append(get_url_myma(I[0][i], audio_names, url_dict))
+            out.append(audio_names[I[0][i]].split('.')[0])
+            out.append(get_url_myma(I[0][i], audio_names, url_dict))
 
     print("Total time : ", time.time() - a)
-    return out
+    return [output_csv]+out
+    
     '''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),