Update app.py

app.py

@@ -31,9 +31,12 @@ from helper import parse_transcription,hindi_to_english,translate_english_to_hin
 def extract_text_from_html(html):
     cleanr = re.compile('<.*?>')
     cleantext = re.sub(cleanr, '', html)
-def conversational_chat(llm_chain,text):    
-    output = llm_chain.predict(human_input=query)
-    return extract_text_from_html(output)
+def conversational_chat(chain,query):    
+    result = chain({"question": query, 
+    "chat_history": st.session_state['history_text']})
+    st.session_state['history_text'].append((query, result["answer"]))
+
+    return result["answer"]
 
 def save_uploaded_file_as_mp3(uploaded_file, output_file_path):
     audio = AudioSegment.from_file(uploaded_file)
@@ -70,12 +73,15 @@ def ui():
         hi_audio,sample_rate2=librosa.load('./good-morning-sir.mp3')
         if 'history' not in st.session_state:
             st.session_state['history'] = []
+            st.session_state['history_text']=[]
     
         if 'generated' not in st.session_state:
             st.session_state['generated'] = [hi_audio]
+            st.session_states['generated_text']=[]
     
         if 'past' not in st.session_state:
             st.session_state['past'] = [good_morining_audio]
+            st.session_states['past_text']=[]
     
         if user_api_key is not None and user_api_key.strip() != "":
             eleven_labs_api_key = st.sidebar.text_input(
@@ -112,8 +118,11 @@ def ui():
                     # hindi_output_file="./Hindi_output_Audio.Mp3"
                     # save_uploaded_file_as_mp3(hindi_out"put_audio,hindi_output_file)
                     st.audio(hindi_output_audio)
+
     
                     st.session_state['past'].append(hindi_input_audio)
+                    st.session_state['past_text'].append(english_input)
+                    st.session_state['past_generated'].append(english_output)
                     st.session_state['generated'].append(hindi_output_audio)
                 
             if 'generated' in st.session_state and st.session_state['generated']:
@@ -128,11 +137,3 @@ def ui():
 
 if __name__ == '__main__':
     ui()
-
-            
-            
-            
-
-
-
-