Update app.py

app.py

@@ -152,71 +152,77 @@ if prompt := st.chat_input("How can I help you today?"):
             message_placeholder.markdown(full_response + "▌")
             message_placeholder.markdown(full_response)
 
-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'] = ''
-
-
-
-
-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:
+    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 = 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['reference'] = '/home/user/app/pdf2image/default_output.png'
+    st.session_state['audio'] = ''
+    
+    
+    
+    
+    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', key='onref')
+    st.session_state.onref = False
+
+    if on_ref:
+        st.session_state.onref = True
+        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', key='ontts')
+    st.session_state.ontts = False
+
+    if on_TTS:
+        st.session_state.ontts = True
+        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})
+    
+