Update app.py

app.py

@@ -35,21 +35,28 @@ def process(prompt, lang):
     print('*************')
     print()
 
-    embed_query  = get_predict(encoder_text, prompt, preprocess_model, model)
-    do_normalize(embed_query)
-    D, I = get_distance(index, embed_query, TOP)
-    #print(I)
-    #print(D)
-    print("----")
-    #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), 
-            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)]
+    try :
+        embed_query  = get_predict(encoder_text, prompt, preprocess_model, model)
+        do_normalize(embed_query)
+        D, I = get_distance(index, embed_query, TOP)
+        #print(I)
+        #print(D)
+        print("----")
+        #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), 
+                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", "",]
 
 inputs = [gr.Textbox(label="Input", value="type your description", max_lines=2), 
             gr.Radio(label="Language", choices=["en"], value="en")]