Update app.py

app.py

@@ -36,7 +36,7 @@ db = get_db_firestore()
 asr = pipeline("automatic-speech-recognition", "facebook/wav2vec2-base-960h")
 
 # create Text Classification pipeline
-#classifier = pipeline("text-classification")
+classifier = pipeline("text-classification")
 
 # create text generator pipeline
 story_gen = pipeline("text-generation", "pranavpsv/gpt2-genre-story-generator")
@@ -73,9 +73,6 @@ def selectall(text):
     docs = db.collection('Text2SpeechSentimentSave').stream()
     doclist=''
     for doc in docs:
-        #docid=doc.id
-        #dict=doc.to_dict()
-        #doclist+=doc.to_dict()
         r=(f'{doc.id} => {doc.to_dict()}')
         doclist += r
     return doclist 
@@ -141,19 +138,19 @@ demo = gr.Blocks()
 with demo:
 
     audio_file = gr.inputs.Audio(source="microphone", type="filepath")
-#    text = gr.Textbox()
-#    label = gr.Label()
-#    saved = gr.Textbox()
-#    savedAll = gr.Textbox()
-#    audio = gr.Audio(label="Output", interactive=False)
+    text = gr.Textbox()
+    label = gr.Label()
+    saved = gr.Textbox()
+    savedAll = gr.Textbox()
+    audio = gr.Audio(label="Output", interactive=False)
     
     b1 = gr.Button("Recognize Speech")
-#    b2 = gr.Button("Classify Sentiment")
-#    b3 = gr.Button("Save Speech to Text")
+    b2 = gr.Button("Classify Sentiment")
+    b3 = gr.Button("Save Speech to Text")
 #    b4 = gr.Button("Retrieve All")
-
-#    b2.click(text_to_sentiment, inputs=text, outputs=label)
- #   b3.click(upsert, inputs=text, outputs=saved)
+    b1.click(speech_to_text, inputs=audio_file, outputs=input_start_text )
+    b2.click(text_to_sentiment, inputs=text, outputs=label)
+    b3.click(upsert, inputs=text, outputs=saved)
  #   b4.click(selectall, inputs=text, outputs=savedAll)
     
     input_story_type = gr.Radio(choices=['superhero', 'action', 'drama', 'horror', 'thriller', 'sci_fi'], value='sci_fi', label="Genre")
@@ -170,9 +167,6 @@ with demo:
     output_interpolation = gr.Video(label="Generated Video")
             
     # Bind functions to buttons
-
-    b1.click(speech_to_text, inputs=audio_file, outputs=input_start_text )
-    
     button_gen_story.click(fn=generate_story, inputs=[input_story_type , input_start_text], outputs=output_generated_story)
     button_gen_images.click(fn=generate_images, inputs=output_generated_story, outputs=output_gallery)
     button_gen_video.click(fn=generate_interpolation, inputs=output_gallery, outputs=output_interpolation)