Update app.py

app.py

@@ -36,20 +36,23 @@ def identity(index, split):
     return ds["prompt"], ds["completion"]
     
 def save(index, language, split, prompt, completion):
-    print("Saving ...")
-    with open("/home/user/app/output.jsonl", "a") as fout :
-        fout.write(
-            json.dumps(
-                {
-                    "prompt" : prompt,
-                    "completion" : completion,
-                    "language" : language,
-                    "index" : index
-                }
-            )+"\n"
-        )
-    next_index = min(1+index, len(samples[split])-1)
-    return next_index, samples[split][next_index]["prompt"], samples[split][next_index]["completion"], "", ""
+    if len(prompt) != 0 and len(completion) != 0 :
+        print("Saving ...")
+        with open("/home/user/app/output.jsonl", "a") as fout :
+            fout.write(
+                json.dumps(
+                    {
+                        "prompt" : prompt,
+                        "completion" : completion,
+                        "language" : language,
+                        "index" : index
+                    }
+                )+"\n"
+            )
+        next_index = min(1+index, len(samples[split])-1)
+        return next_index, samples[split][next_index]["prompt"], samples[split][next_index]["completion"], "", ""
+    else :
+        return index, samples[split][index]["prompt"], samples[split][index]["completion"], "", ""
 
 with gr.Blocks(analytics_enabled=False, css=custom_css) as demo:
     gr.HTML("""<h1 align="center">MT💫</h1>""")