Update app.py

app.py

@@ -1,32 +1,81 @@
-import numpy as np
+import os
+import warnings
 import gradio as gr
+import re
 
-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")
 
-    img_input.change(sepia, [img_input, strength], img_output)
-    strength.change(sepia, [img_input, strength], img_output)
+HF_TOKEN = os.getenv('HW_TOKEN')
+hf_writer = gr.HuggingFaceDatasetSaver(HF_TOKEN, "save_audio")
+
+
+cur_line=0
+
+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)
     
-    # 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)
+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]
+
 
-demo.launch()
+
+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()