Update app.py

app.py

@@ -148,74 +148,74 @@ if prompt := st.chat_input("How can I help you today?"):
         result = qa_chain(prompt)
         output = [result['result']]
 
-    def generate_pdf():
-        page_number = int(result['source_documents'][0].metadata['page'])
-        doc = fitz.open(str(result['source_documents'][0].metadata['source']))
-        text = str(result['source_documents'][0].page_content)
-        if text != '':
-            for page in doc:
-                text_instances = page.search_for(text)
-                for inst in text_instances:
-                    highlight = page.add_highlight_annot(inst)
-                    highlight.update()
-        doc.save("/home/user/app/pdf2image/output.pdf", garbage=4, deflate=True, clean=True)
-
-        def pdf_page_to_image(pdf_file, page_number, output_image):
-            pdf_document = fitz.open(pdf_file)
-            page = pdf_document[page_number]
-            dpi = 300  # You can adjust this as needed
-            pix = page.get_pixmap(matrix=fitz.Matrix(dpi / 100, dpi / 100))
-            pix.save(output_image, "png")
-            pdf_document.close()
-        pdf_page_to_image('/home/user/app/pdf2image/output.pdf', page_number, '/home/user/app/pdf2image/output.png')
-        #image = Image.open('/home/user/app/pdf2image/output.png')
-        #message_placeholder.image(image)
-        #st.session_state.reference = True
-
-
-
-    def generate_audio():
+def generate_pdf():
+    page_number = int(result['source_documents'][0].metadata['page'])
+    doc = fitz.open(str(result['source_documents'][0].metadata['source']))
+    text = str(result['source_documents'][0].page_content)
+    if text != '':
+        for page in doc:
+            text_instances = page.search_for(text)
+            for inst in text_instances:
+                highlight = page.add_highlight_annot(inst)
+                highlight.update()
+    doc.save("/home/user/app/pdf2image/output.pdf", garbage=4, deflate=True, clean=True)
+
+    def pdf_page_to_image(pdf_file, page_number, output_image):
+        pdf_document = fitz.open(pdf_file)
+        page = pdf_document[page_number]
+        dpi = 300  # You can adjust this as needed
+        pix = page.get_pixmap(matrix=fitz.Matrix(dpi / 100, dpi / 100))
+        pix.save(output_image, "png")
+        pdf_document.close()
+    pdf_page_to_image('/home/user/app/pdf2image/output.pdf', page_number, '/home/user/app/pdf2image/output.png')
+    #image = Image.open('/home/user/app/pdf2image/output.png')
+    #message_placeholder.image(image)
+    #st.session_state.reference = True
+
+
+
+def generate_audio():
+    with open('/home/user/app/audio/audio.mp3','wb') as sound_file:
+        tts = gTTS(result['result'], lang='en', tld='co.in')
+        tts.write_to_fp(sound_file)
+    sound = AudioSegment.from_mp3("/home/user/app/audio/audio.mp3")
+    sound.export("/home/user/app/audio/audio.wav", format="wav")
+
+st.session_state['reference'] = '/home/user/app/pdf2image/default_output.png'
+st.session_state['audio'] = ''
+
+for item in output:
+    full_response += item
+    message_placeholder.markdown(full_response + "▌")
+    message_placeholder.markdown(full_response)
+
+
+if "reference" not in st.session_state:
+    st.session_state.reference = False
+if "audio" not in st.session_state:
+    st.session_state.audio = False
+
+
+
+#choice = st.radio("References and TTS",["Reference" , 'TTS'])
+on_ref = st.toggle('Reference')
+if on_ref:
+    with st.sidebar:
+        generate_pdf()
+        st.session_state['reference'] = '/home/user/app/pdf2image/output.png'
+        st.image(st.session_state['reference'])
+
+on_TTS = st.toggle('TTS')
+if on_TTS:
+    with st.sidebar:
         with open('/home/user/app/audio/audio.mp3','wb') as sound_file:
-            tts = gTTS(result['result'], lang='en', tld='co.in')
+            tts = gTTS(result['result'], lang='en', tld = 'co.in')
             tts.write_to_fp(sound_file)
         sound = AudioSegment.from_mp3("/home/user/app/audio/audio.mp3")
         sound.export("/home/user/app/audio/audio.wav", format="wav")
+        st.session_state['audio'] = '/home/user/app/audio/audio.wav'
+        st.audio(st.session_state['audio'])
 
-    st.session_state['reference'] = '/home/user/app/pdf2image/default_output.png'
-    st.session_state['audio'] = ''
-    
-    for item in output:
-        full_response += item
-        message_placeholder.markdown(full_response + "▌")
-        message_placeholder.markdown(full_response)
-
-
-    if "reference" not in st.session_state:
-        st.session_state.reference = False
-    if "audio" not in st.session_state:
-        st.session_state.audio = False
-
-
-    
-    #choice = st.radio("References and TTS",["Reference" , 'TTS'])
-    on_ref = st.toggle('Reference')
-    if on_ref:
-        with st.sidebar:
-            generate_pdf()
-            st.session_state['reference'] = '/home/user/app/pdf2image/output.png'
-            st.image(st.session_state['reference'])
-    
-    on_TTS = st.toggle('TTS')
-    if on_TTS:
-        with st.sidebar:
-            with open('/home/user/app/audio/audio.mp3','wb') as sound_file:
-                tts = gTTS(result['result'], lang='en', tld = 'co.in')
-                tts.write_to_fp(sound_file)
-            sound = AudioSegment.from_mp3("/home/user/app/audio/audio.mp3")
-            sound.export("/home/user/app/audio/audio.wav", format="wav")
-            st.session_state['audio'] = '/home/user/app/audio/audio.wav'
-            st.audio(st.session_state['audio'])
-
-    st.session_state.messages.append({"role": "assistant", "content": full_response})
+st.session_state.messages.append({"role": "assistant", "content": full_response})