upgrade gradio

README.md

@@ -4,7 +4,7 @@ emoji: 🔁
 colorFrom: indigo
 colorTo: pink
 sdk: gradio
-sdk_version: 4.40.0
+sdk_version: 5.9.0
 app_file: app.py
 pinned: false
 ---

app.py

@@ -149,13 +149,28 @@ def generate_random(sent1, count, approx):
     if sent1 == '':
         raise gr.Error('Please input a source text.')
     preds, interpolations = [], []
+    orig_active_indices = shared_state.active_indices
+    shared_state.active_indices = set(range(len(lng_names)))
     for c in range(count):
         idx = np.random.randint(0, len(ling_collection))
         ling_ex = ling_collection[idx]
         shared_state.target = ling_ex.copy()
-        pred, interpolation = generate_with_feedback(sent1, approx)
+        success = False
+        patience = 0
+        while not success:
+            pred, interpolation = generate_with_feedback(sent1, approx)[:2]
+            if pred not in preds:
+                success = True
+            elif patience < 3:
+                add_to_target()
+                patience += 1
+            else:
+                idx = np.random.randint(0, len(ling_collection))
+                ling_ex = ling_collection[idx]
+                shared_state.target = ling_ex.copy()
         preds.append(pred)
         interpolations.append(interpolation)
+    shared_state.active_indices = orig_active_indices
     return '\n***\n'.join(preds), '\n***\n'.join(interpolations)
 
 def estimate_gen(sent1, sent2, approx):
@@ -172,8 +187,8 @@ def estimate_gen(sent1, sent2, approx):
     ling_pred = round_ling(ling_pred)
     shared_state.target = ling_pred.copy()
     
-    gen = generate_with_feedback(sent1, approx)
-    return gen[0], gen[1], [gr.update(value=val) for val in shared_state.target]
+    gen = generate_with_feedback(sent1, approx)[:2]
+    return gen + [gr.update(value=val) for val in shared_state.target]
 
 def estimate_tgt(sent2, ling_dict, approx):
     if 'approximate' in approx: