Update app.py

app.py

@@ -86,20 +86,20 @@ def save_results(text, label):
     # Save to JSON
     if not os.path.exists(RESULTS_JSON):
         with open(RESULTS_JSON, "w") as f:
-            json.dump([], f)
+            json.dump([], f)  # Initialize an empty list if file doesn't exist
     with open(RESULTS_JSON, "r+") as f:
-        content = json.load(f)
-        content.append(data)
-        f.seek(0)
-        json.dump(content, f, indent=4)
+        content = json.load(f)  # Load current content
+        content.append(data)  # Append new data
+        f.seek(0)  # Go back to the start of the file
+        json.dump(content, f, indent=4)  # Save updated content
 
     # Save to CSV
     file_exists = os.path.exists(RESULTS_CSV)
     with open(RESULTS_CSV, "a", newline="") as f:
         writer = csv.DictWriter(f, fieldnames=["text", "label"])
         if not file_exists:
-            writer.writeheader()
-        writer.writerow(data)
+            writer.writeheader()  # Write header if file is new
+        writer.writerow(data)  # Append new data
 
 # Gradio Interface
 image_input = gr.Image()