Update app.py

app.py

@@ -68,15 +68,16 @@ def csv(segments):
       output += f"{segment['start']},{segment['end']},{segment['text']}\n"
     return output
 def transcribe(path,lang,size):
+    yield (None,[("","Transcribing Video...")])
     #if size != current_size:
     loaded_model = whisper.load_model(size)
     current_size = size
     results = loaded_model.transcribe(path, language=lang)
     subs = ".csv"
     if subs == "None":
-      return results["text"]
+      yield results["text"],[("","Transcription Complete...")]
     elif subs == ".csv":
-      return csv(results["segments"])
+      yield csv(results["segments"]),[("","Transcription Complete...")]
 
 
 def format_prompt(message, history):
@@ -227,8 +228,8 @@ def compress_data_og(c, instruct, history):
 
 
 
-def summarize(inp,history,report_check,sum_mem_check,data=None):
-
+def summarize(inp,history,mem_check,data=None):
+    
     json_box=[]
     error_box=""
     json_out={}
@@ -236,7 +237,7 @@ def summarize(inp,history,report_check,sum_mem_check,data=None):
     if inp == "":
         inp = "Process this data"
     history.clear()
-    history = [(inp,"Working on it...")] 
+    history = [(inp,"Summarizing Transcription...")] 
     yield "",history,error_box,json_box
 
     if data != "Error" and data != "" and data != None:
@@ -250,29 +251,24 @@ def summarize(inp,history,report_check,sum_mem_check,data=None):
             if i == " " or i=="," or i=="\n":
                 c +=1
         print (f'c:: {c}')
-        if sum_mem_check=="Memory":
-            #save_memory(inp,out)
-            rawp = "Complete"
-        if sum_mem_check=="Summary":
-            json_out = compress_data(c,inp,out)  
-    
-            out = str(json_out)
-            print (out)
-            if report_check:
-                rl = len(out)
-                print(f'rl:: {rl}')
-                c=1
-                for i in str(out):
-                    if i == " " or i=="," or i=="\n":
-                        c +=1
-                print (f'c2:: {c}')
-                rawp = compress_data_og(c,inp,out)
-                history.clear()
-                history.append((inp,rawp))                
-            else:
-                rawp = out
-                history.clear()
-                history.append((inp,rawp))
+        json_out = compress_data(c,inp,out)  
+        history = [(inp,"Generating Report...")] 
+        yield "", history,error_box,json_out
+
+        out = str(json_out)
+        print (out)
+        if report_check:
+        rl = len(out)
+        print(f'rl:: {rl}')
+        c=1
+        for i in str(out):
+            if i == " " or i=="," or i=="\n":
+                c +=1
+        print (f'c2:: {c}')
+        rawp = compress_data_og(c,inp,out)
+        history.clear()
+        history.append((inp,rawp))                
+
         yield "", history,error_box,json_out
     else:
         rawp = "Provide a valid data source"
@@ -286,7 +282,7 @@ def clear_fn():
     return "",[(None,None)]
 
 with gr.Blocks() as app:
-    gr.HTML("""<center><h1>Mixtral 8x7B TLDR Summarizer + Web</h1><h3>Summarize Data of unlimited length</h3>""")
+    gr.HTML("""<center><h1>Video Summarizer</h1><h3>Mixtral 8x7B + Whisper</h3>""")
     with gr.Row():
         with gr.Column():
             with gr.Row():
@@ -301,8 +297,7 @@ with gr.Blocks() as app:
         with gr.Column(scale=3):
             prompt=gr.Textbox(label = "Instructions (optional)")
         with gr.Column(scale=1):
-            report_check=gr.Checkbox(label="Return Report", value=True)
-            sum_mem_check=gr.Radio(label="Output",choices=["Summary","Memory"])
+            mem_check=gr.Checkbox(label="Memory", value=False)
             button=gr.Button()
         
         #models_dd=gr.Dropdown(choices=[m for m in return_list],interactive=True)
@@ -320,6 +315,6 @@ with gr.Blocks() as app:
     url_btn.click(dl,[inp_url,vid],vid)
     trans_btn.click(transcribe,[vid,lang,sz],trans)
     clear_btn.click(clear_fn,None,[prompt,chatbot])
-    go=button.click(summarize,[prompt,chatbot,report_check,sum_mem_check,trans],[prompt,chatbot,e_box,json_out])
+    go=button.click(transcribe,[vid,lang,sz],[trans,chatbot]).then(summarize,[prompt,chatbot,mem_check,trans],[prompt,chatbot,e_box,json_out])
     stop_button.click(None,None,None,cancels=[go])
 app.queue(default_concurrency_limit=20).launch(show_api=False)