Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -62,65 +62,65 @@ def ui():
|
|
62 |
template=template
|
63 |
)
|
64 |
|
65 |
-
|
66 |
-
|
67 |
-
|
68 |
-
|
69 |
-
|
70 |
-
|
71 |
-
|
72 |
-
|
73 |
-
|
74 |
-
|
75 |
-
|
76 |
-
|
77 |
-
|
78 |
-
|
79 |
-
|
80 |
-
|
81 |
-
|
82 |
-
|
83 |
-
|
84 |
-
|
85 |
-
|
86 |
-
|
87 |
-
|
88 |
-
|
89 |
-
#container for the chat history
|
90 |
-
response_container = st.container()
|
91 |
-
#container for the user's text input
|
92 |
-
container = st.container()
|
93 |
-
|
94 |
-
with container:
|
95 |
-
with st.form(key='my_form', clear_on_submit=True):
|
96 |
-
audio_file = audiorecorder("Click to record", "Recording...")
|
97 |
-
wav_file = open("./output_audio.mp3", "wb")
|
98 |
-
wav_file.write(audio.tobytes())
|
99 |
-
submit_button = st.form_submit_button(label='Send')
|
100 |
-
if submit_button and audio_file:
|
101 |
-
output_file_path = "./output_audio.mp3"
|
102 |
-
# save_uploaded_file_as_mp3(audio_file,output_file_path )
|
103 |
-
hindi_input_audio,sample_rate=torchaudio.load(output_file_path)
|
104 |
-
#applying the audio recognition
|
105 |
-
hindi_transcription=parse_transcription('./output_audio.mp3')
|
106 |
-
st.success(f"Audio file saved as {output_file_path}")
|
107 |
-
#convert hindi to english
|
108 |
-
english_input=hindi_to_english(hindi_transcription)
|
109 |
-
#feeding the input to the LLM
|
110 |
-
english_output = conversational_chat(english_input)
|
111 |
-
#converting english to hindi
|
112 |
-
hin_output=translate_english_to_hindi(english_output)
|
113 |
-
#getting the hindi_tts
|
114 |
-
hindi_output_audio=hindi_tts(hin_output)
|
115 |
-
|
116 |
-
st.session_state['past'].append(hindi_input_audio)
|
117 |
-
st.session_state['generated'].append(hindi_output_audio)
|
118 |
|
119 |
-
|
120 |
-
|
121 |
-
|
122 |
-
|
123 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
124 |
|
125 |
if __name__ == '__main__':
|
126 |
ui()
|
|
|
62 |
template=template
|
63 |
)
|
64 |
|
65 |
+
llm_chain = LLMChain(
|
66 |
+
llm = ChatOpenAI(temperature=0.0,model_name='gpt-3.5-turbo'),
|
67 |
+
prompt=prompt,
|
68 |
+
verbose=True,
|
69 |
+
memory=ConversationBufferWindowMemory(k=2)
|
70 |
+
)
|
71 |
+
|
72 |
+
if 'history' not in st.session_state:
|
73 |
+
st.session_state['history'] = []
|
74 |
+
|
75 |
+
if 'generated' not in st.session_state:
|
76 |
+
st.session_state['generated'] = ["Hello ! Ask me anything about " + " π€"]
|
77 |
+
|
78 |
+
if 'past' not in st.session_state:
|
79 |
+
st.session_state['past'] = ["Hey ! π"]
|
80 |
+
|
81 |
+
if user_api_key is not None and user_api_key.strip() != "":
|
82 |
+
eleven_labs_api_key = st.sidebar.text_input(
|
83 |
+
label="#### Your Eleven Labs API key π",
|
84 |
+
placeholder="Paste your Eleven Labs API key",
|
85 |
+
type="password")
|
86 |
+
|
87 |
+
set_api_key(user_api_key)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
88 |
|
89 |
+
#container for the chat history
|
90 |
+
response_container = st.container()
|
91 |
+
#container for the user's text input
|
92 |
+
container = st.container()
|
93 |
+
|
94 |
+
with container:
|
95 |
+
with st.form(key='my_form', clear_on_submit=True):
|
96 |
+
audio_file = audiorecorder("Click to record", "Recording...")
|
97 |
+
wav_file = open("./output_audio.mp3", "wb")
|
98 |
+
wav_file.write(audio.tobytes())
|
99 |
+
submit_button = st.form_submit_button(label='Send')
|
100 |
+
if submit_button and audio_file:
|
101 |
+
output_file_path = "./output_audio.mp3"
|
102 |
+
# save_uploaded_file_as_mp3(audio_file,output_file_path )
|
103 |
+
hindi_input_audio,sample_rate=torchaudio.load(output_file_path)
|
104 |
+
#applying the audio recognition
|
105 |
+
hindi_transcription=parse_transcription('./output_audio.mp3')
|
106 |
+
st.success(f"Audio file saved as {output_file_path}")
|
107 |
+
#convert hindi to english
|
108 |
+
english_input=hindi_to_english(hindi_transcription)
|
109 |
+
#feeding the input to the LLM
|
110 |
+
english_output = conversational_chat(english_input)
|
111 |
+
#converting english to hindi
|
112 |
+
hin_output=translate_english_to_hindi(english_output)
|
113 |
+
#getting the hindi_tts
|
114 |
+
hindi_output_audio=hindi_tts(hin_output)
|
115 |
+
|
116 |
+
st.session_state['past'].append(hindi_input_audio)
|
117 |
+
st.session_state['generated'].append(hindi_output_audio)
|
118 |
+
|
119 |
+
if st.session_state['generated']:
|
120 |
+
with response_container:
|
121 |
+
for i in range(len(st.session_state['generated'])):
|
122 |
+
st.audio(st.session_state["past"][i],format='audio/wav')
|
123 |
+
st.audio(st.session_state["generated"][i],format='audio/wav')
|
124 |
|
125 |
if __name__ == '__main__':
|
126 |
ui()
|