Update app.py

app.py

@@ -1,5 +1,6 @@
 import gradio as gr
 import json
+import os
 
 class JSONLViewer:
     def __init__(self, file_path):
@@ -9,8 +10,11 @@ class JSONLViewer:
         self.load_data()
 
     def load_data(self):
-        with open(self.file_path, 'r', encoding='utf-8') as file:
-            self.data = [json.loads(line) for line in file]
+        if os.path.exists(self.file_path):
+            with open(self.file_path, 'r', encoding='utf-8') as file:
+                self.data = [json.loads(line) for line in file]
+        else:
+            self.data = []
 
     def save_data(self):
         with open(self.file_path, 'w', encoding='utf-8') as file:
@@ -70,7 +74,7 @@ def on_rejected():
     record = viewer.get_current_record()
     return update_ui(record)
 
-with gr.Blocks() as demo:
+with gr.Blocks(css="button.ok-button { background-color: #4CAF50 !important; }") as demo:
     instruction = gr.Textbox(label="INSTRUCTION", lines=8, max_lines=8)
     chosen = gr.Textbox(label="CHOSEN", lines=8, max_lines=8)
     rejected = gr.Textbox(label="REJECTED", lines=8, max_lines=8)
@@ -81,8 +85,8 @@ with gr.Blocks() as demo:
     with gr.Row():
         prev_btn = gr.Button("PREV")
         next_btn = gr.Button("NEXT")
-        ok_btn = gr.Button("OK", variant="primary")  # Zielony przycisk
-        rejected_btn = gr.Button("REJECTED", variant="stop")  # Czerwony przycisk
+        ok_btn = gr.Button("OK", elem_classes="ok-button")
+        rejected_btn = gr.Button("REJECTED", variant="stop")
 
     prev_btn.click(on_prev, outputs=[instruction, chosen, rejected, chosen_score, rejected_score, status])
     next_btn.click(on_next, outputs=[instruction, chosen, rejected, chosen_score, rejected_score, status])
@@ -100,4 +104,5 @@ with gr.Blocks() as demo:
         rejected_score.value = init_rejected_score
         status.value = init_status
 
-demo.launch()
+if __name__ == "__main__":
+    demo.launch()