Update app.py

app.py

@@ -2,80 +2,37 @@ import os
 import warnings
 import gradio as gr
 import re
+import numpy as np
+
+#HF_TOKEN = os.getenv('HW_TOKEN')
+#hf_writer = gr.HuggingFaceDatasetSaver(HF_TOKEN, "save_audio")
+
+def sepia(input_img, strength):
+    sepia_filter = strength * np.array(
+        [[0.393, 0.769, 0.189], [0.349, 0.686, 0.168], [0.272, 0.534, 0.131]]
+    ) + (1-strength) * np.identity(3)
+    sepia_img = input_img.dot(sepia_filter.T)
+    sepia_img /= sepia_img.max()
+    return sepia_img
+
+callback = gr.CSVLogger()
+
+with gr.Blocks() as demo:
+    with gr.Row():
+        with gr.Column():
+            img_input = gr.Image()
+            strength = gr.Slider(0, 1, 0.5)
+        img_output = gr.Image()
+    with gr.Row():
+        btn = gr.Button("Flag")
+        
+    # This needs to be called at some point prior to the first call to callback.flag()
+    callback.setup([img_input, strength, img_output], "flagged_data_points")
 
-
-HF_TOKEN = os.getenv('HW_TOKEN')
-hf_writer = gr.HuggingFaceDatasetSaver(HF_TOKEN, "save_audio")
-
-
-cur_line=-1
-
-def readFile():
-    f=open('prompt.txt')
-    line_num=0
-    lines=f.readlines()
-    line_num = len(lines)
-    return  line_num,lines
-
-totlines,file_content=readFile()
-
-#callback = gr.CSVLogger()
-def recordAndsave(text,audio):
-    #print (next)
-    print (text)
-    global totlines
-    print(totlines)
-    global cur_line
-    if cur_line<totlines-1:
-        cur_line+=1
-        global file_content
-        print (cur_line)
-    return file_content[cur_line].strip()
-    #return None
-
-    #print (previous)
+    img_input.change(sepia, [img_input, strength], img_output)
+    strength.change(sepia, [img_input, strength], img_output)
     
-def readPromt():
-    global cur_line
-    cur_line+=1
-    global file_content
-    print (cur_line)
-    return file_content[cur_line]
-    
-def readNext():
-    
-    global totlines
-    print(totlines)
-    global cur_line
-    if cur_line<totlines-1:
-        cur_line+=1
-        global file_content
-        print (cur_line)
-    return [file_content[cur_line],None]
-
-def readPrevious():
-    global cur_line
-    if cur_line>=0:
-        cur_line-=1
-    #cur_line=current_line
-        global file_content
-        print (cur_line)
-    return [file_content[cur_line],None]
-
+    # We can choose which components to flag -- in this case, we'll flag all of them
+    btn.click(lambda *args: callback.flag(args), [img_input, strength, img_output], None, preprocess=False)
 
-
-demo=gr.Interface(
-    fn=recordAndsave, 
-   
-
-    inputs=[
-        gr.Audio(sources=["microphone","upload"], type="filepath"),
-       
-    ],
-    outputs=[
-        gr.Textbox(readPromt(),label="Prompt")
-    ],
-        
-        allow_flagging="manual",
-        flagging_callback=hf_writer
-    ).launch()
+demo.launch()