Update app.py

app.py

@@ -188,6 +188,9 @@ def get_message_history():
 docs = load_data()
 qa_chain = load_model(docs)
 
+
+
+
 print('2')
 print(time.time())
 if prompt := st.chat_input("How can I help you today?"):
@@ -206,9 +209,7 @@ if prompt := st.chat_input("How can I help you today?"):
         result = qa_chain(prompt)
         print('4')
         print(time.time())
-        sound_file = BytesIO()
-        tts = gTTS(result['result'], lang='en')
-        tts.write_to_fp(sound_file)
+        
         output = [result['result']]
 
     for item in output:
@@ -220,116 +221,60 @@ if prompt := st.chat_input("How can I help you today?"):
 
     print('5')
     print(time.time())
-    ### READ IN 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:
-            ### SEARCH
-            text_instances = page.search_for(text)
-
-            ### HIGHLIGHT
-            for inst in text_instances:
-                highlight = page.add_highlight_annot(inst)
-                highlight.update()
-
-    ### OUTPUT
-    doc.save("/home/user/app/pdf2image/output.pdf", garbage=4, deflate=True, clean=True)
-
-    # pdf_to_open = repr(result['source_documents'][0].metadata['source'])
-
-    def pdf_page_to_image(pdf_file, page_number, output_image):
-        # Open the PDF file
-        pdf_document = fitz.open(pdf_file)
-
-        # Get the specific page
-        page = pdf_document[page_number]
-
-        # Define the image DPI (dots per inch)
-        dpi = 300  # You can adjust this as needed
-
-        # Convert the page to an image
-        pix = page.get_pixmap(matrix=fitz.Matrix(dpi / 100, dpi / 100))
-
-        # Save the image as a PNG file
-        pix.save(output_image, "png")
-
-        # Close the PDF file
-        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')
-    st.image(image)
-    st.audio(sound_file)
-
-    # if 'clickedR' not in st.session_state:
-    #     st.session_state.clickedR = False
-
-    # def click_buttonR():
-    #     st.session_state.clickedR = True
-    #     if st.session_state.clickedR:
-    #         message_placeholder.markdown(full_response+repr(result['source_documents'][0]))
-
-    # ref = st.button('References', on_click = click_buttonR)
+    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:
+                ### SEARCH
+                text_instances = page.search_for(text)
+    
+                ### HIGHLIGHT
+                for inst in text_instances:
+                    highlight = page.add_highlight_annot(inst)
+                    highlight.update()
+    
+        ### OUTPUT
+        doc.save("/home/user/app/pdf2image/output.pdf", garbage=4, deflate=True, clean=True)
+    
+        # pdf_to_open = repr(result['source_documents'][0].metadata['source'])
+    
+        def pdf_page_to_image(pdf_file, page_number, output_image):
+            # Open the PDF file
+            pdf_document = fitz.open(pdf_file)
+    
+            # Get the specific page
+            page = pdf_document[page_number]
+    
+            # Define the image DPI (dots per inch)
+            dpi = 300  # You can adjust this as needed
     
-    # if 'clickedA' not in st.session_state:
-    #     st.session_state.clickedA = False
-
-    # def click_buttonA():
-    #     st.session_state.clickedA = True
-    #     if st.session_state.clickedA:
-    #         sound_file = BytesIO()
-    #         tts = gTTS(result['result'], lang='en')
-    #         tts.write_to_fp(sound_file)
-    #         st.audio(sound_file)  
-
-
-    # ref = st.button(':speaker:', on_click = click_buttonA)
-
-  
-
-
-
-    #st.session_state.clickedR = False
-
-    # #if ref:
-    # message_placeholder.markdown(full_response+repr(result['source_documents'][0]))
-    # #if sound:
-    # sound_file = BytesIO()
-    # tts = gTTS(result['result'], lang='en')
-    # tts.write_to_fp(sound_file)
-    # html_string = """
-    # <audio controls autoplay>
-    #   <source src="/content/sound_file" type="audio/wav">
-    # </audio>
-    # """
-    # message_placeholder.markdown(html_string, unsafe_allow_html=True)  # will display a st.audio with the sound you specified in the "src" of the html_string and autoplay it
-    # #time.sleep(5)  # wait for 2 seconds to finish the playing of the audio
-    #response_sentiment = st.radio(
-    #    "How was the Assistant's response?",
-    #    ["😁", "😕", "😢"],
-    #    key="response_sentiment",
-    #    disabled=st.session_state.disabled,
-    #    horizontal=True,
-    #    index=1,
-    #    help="This helps us improve the model.",
-    #    # hide the radio button on click
-    #   on_change=on_select(),
-    #)
-    #logger.info(f"{user_session_id} | {full_response} | {response_sentiment}")
-
-    # # Logging to FastAPI Endpoint
-    # headers = {"Authorization": f"Bearer {secret_token}"}
-    # log_data = {"log": f"{user_session_id} | {full_response} | {response_sentiment}"}
-    # response = requests.post(fastapi_endpoint, json=log_data, headers=headers, timeout=10)
-    # if response.status_code == 200:
-    #     logger.info("Query logged successfully")
+            # Convert the page to an image
+            pix = page.get_pixmap(matrix=fitz.Matrix(dpi / 100, dpi / 100))
+    
+            # Save the image as a PNG file
+            pix.save(output_image, "png")
+    
+            # Close the PDF file
+            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')
+        st.image(image)
 
+    def generate_audio():
+        sound_file = BytesIO()
+        tts = gTTS(result['result'], lang='en')
+        tts.write_to_fp(sound_file)
+        st.audio(sound_file)
+    
+    st.button(':speaker:', type='primary',on_click=generate_audio)
+    st.button('Reference',type='primary',on_click=generate_pdf)
     st.session_state.messages.append({"role": "assistant", "content": full_response})