Update app.py

app.py

@@ -100,10 +100,16 @@ classifier = pipeline("text-classification")
 
 def speech_to_text(speech):
     text = asr(speech)["text"]
+    
+    rMem = AIMemory("STT", text) 
+    
     return text
 
 def text_to_sentiment(text):
     sentiment = classifier(text)[0]["label"]
+    
+    rMem = AIMemory(text, sentiment) 
+
     return sentiment 
 
 def upsert(text):
@@ -112,6 +118,9 @@ def upsert(text):
     doc_ref.set({u'firefield': 'Recognize Speech', u'first': 'https://huggingface.co/spaces/awacke1/TTS-STT-Blocks/', u'last': text, u'born': date_time,})
     saved = select('TTS-STT', date_time)
     # check it here:  https://console.firebase.google.com/u/0/project/clinical-nlp-b9117/firestore/data/~2FStreamlitSpaces
+    
+    rMem = AIMemory(text, saved) 
+    
     return saved
       
 def select(collection, document):
@@ -137,6 +146,9 @@ def tts(text: str, model_name: str):
     wavs = synthesizer.tts(text)
     with tempfile.NamedTemporaryFile(suffix=".wav", delete=False) as fp:
         synthesizer.save_wav(wavs, fp)
+        
+        rMem = AIMemory("TTS", text + model_name) 
+    
         return fp.name
 
 demo = gr.Blocks()