Update app.py
Browse files
app.py
CHANGED
@@ -24,7 +24,6 @@ from PIL import Image
|
|
24 |
from langchain.vectorstores import FAISS
|
25 |
import transformers
|
26 |
from pydub import AudioSegment
|
27 |
-
from streamlit_extras.stateful_button import button
|
28 |
|
29 |
user_session_id = uuid.uuid4()
|
30 |
|
@@ -169,8 +168,6 @@ if prompt := st.chat_input("How can I help you today?"):
|
|
169 |
pix.save(output_image, "png")
|
170 |
pdf_document.close()
|
171 |
pdf_page_to_image('/home/user/app/pdf2image/output.pdf', page_number, '/home/user/app/pdf2image/output.png')
|
172 |
-
pdfPath = '/home/user/app/pdf2image/output.png'
|
173 |
-
return pdfPath
|
174 |
#image = Image.open('/home/user/app/pdf2image/output.png')
|
175 |
#message_placeholder.image(image)
|
176 |
#st.session_state.reference = True
|
@@ -183,23 +180,38 @@ if prompt := st.chat_input("How can I help you today?"):
|
|
183 |
tts.write_to_fp(sound_file)
|
184 |
sound = AudioSegment.from_mp3("/home/user/app/audio/audio.mp3")
|
185 |
sound.export("/home/user/app/audio/audio.wav", format="wav")
|
186 |
-
|
187 |
-
|
|
|
188 |
|
189 |
for item in output:
|
190 |
full_response += item
|
191 |
message_placeholder.markdown(full_response + "▌")
|
192 |
message_placeholder.markdown(full_response)
|
193 |
|
194 |
-
# if button('Audio', key='audio'):
|
195 |
-
# ap = generate_audio()
|
196 |
-
# st.sidebar.audio(ap)
|
197 |
|
198 |
-
if
|
199 |
-
|
200 |
-
|
|
|
|
|
|
|
|
|
|
|
201 |
|
|
|
|
|
|
|
|
|
202 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
203 |
|
204 |
st.session_state.messages.append({"role": "assistant", "content": full_response})
|
205 |
|
|
|
24 |
from langchain.vectorstores import FAISS
|
25 |
import transformers
|
26 |
from pydub import AudioSegment
|
|
|
27 |
|
28 |
user_session_id = uuid.uuid4()
|
29 |
|
|
|
168 |
pix.save(output_image, "png")
|
169 |
pdf_document.close()
|
170 |
pdf_page_to_image('/home/user/app/pdf2image/output.pdf', page_number, '/home/user/app/pdf2image/output.png')
|
|
|
|
|
171 |
#image = Image.open('/home/user/app/pdf2image/output.png')
|
172 |
#message_placeholder.image(image)
|
173 |
#st.session_state.reference = True
|
|
|
180 |
tts.write_to_fp(sound_file)
|
181 |
sound = AudioSegment.from_mp3("/home/user/app/audio/audio.mp3")
|
182 |
sound.export("/home/user/app/audio/audio.wav", format="wav")
|
183 |
+
|
184 |
+
st.session_state['reference'] = '/home/user/app/pdf2image/default_output.png'
|
185 |
+
st.session_state['audio'] = ''
|
186 |
|
187 |
for item in output:
|
188 |
full_response += item
|
189 |
message_placeholder.markdown(full_response + "▌")
|
190 |
message_placeholder.markdown(full_response)
|
191 |
|
|
|
|
|
|
|
192 |
|
193 |
+
if "reference" not in st.session_state:
|
194 |
+
st.session_state.reference = False
|
195 |
+
if "audio" not in st.session_state:
|
196 |
+
st.session_state.audio = False
|
197 |
+
|
198 |
+
|
199 |
+
with st.sidebar:
|
200 |
+
choice = st.radio("References",["Reference"], index=None,)
|
201 |
|
202 |
+
if choice == 'Reference':
|
203 |
+
generate_pdf()
|
204 |
+
st.session_state['reference'] = '/home/user/app/pdf2image/output.png'
|
205 |
+
st.image(st.session_state['reference'])
|
206 |
|
207 |
+
# if choice == 'TTS':
|
208 |
+
# with open('/home/user/app/audio/audio.mp3','wb') as sound_file:
|
209 |
+
# tts = gTTS(result['result'], lang='en', tld = 'co.in')
|
210 |
+
# tts.write_to_fp(sound_file)
|
211 |
+
# sound = AudioSegment.from_mp3("/home/user/app/audio/audio.mp3")
|
212 |
+
# sound.export("/home/user/app/audio/audio.wav", format="wav")
|
213 |
+
# st.session_state['audio'] = '/home/user/app/audio/audio.wav'
|
214 |
+
# st.audio(st.session_state['audio'])
|
215 |
|
216 |
st.session_state.messages.append({"role": "assistant", "content": full_response})
|
217 |
|