eagle0504 commited on
Commit
1f08ed4
1 Parent(s): 87ec466

app updated

Browse files
Files changed (2) hide show
  1. app.py +3 -0
  2. helper/utils.py +26 -0
app.py CHANGED
@@ -59,6 +59,9 @@ with st.sidebar:
59
  # Clear button
60
  clear_button = st.sidebar.button("Clear Conversation", key="clear")
61
 
 
 
 
62
 
63
  # Initialize chat history
64
  if "messages" not in st.session_state:
 
59
  # Clear button
60
  clear_button = st.sidebar.button("Clear Conversation", key="clear")
61
 
62
+ # Display pdf
63
+ displayPDF(uploaded_files)
64
+
65
 
66
  # Initialize chat history
67
  if "messages" not in st.session_state:
helper/utils.py CHANGED
@@ -1,6 +1,8 @@
1
  import os
2
  from typing import Any, Dict, List, Tuple, Union
3
 
 
 
4
  import numpy as np
5
  import pandas as pd
6
  import PyPDF2
@@ -45,6 +47,30 @@ def read_and_textify(
45
  return [text_list, sources_list]
46
 
47
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
48
  client = OpenAI(api_key=os.environ["OPENAI_API_KEY"])
49
 
50
 
 
1
  import os
2
  from typing import Any, Dict, List, Tuple, Union
3
 
4
+ import base64
5
+ import streamlit as st
6
  import numpy as np
7
  import pandas as pd
8
  import PyPDF2
 
47
  return [text_list, sources_list]
48
 
49
 
50
+ def displayPDF(file: str) -> None:
51
+ """
52
+ This function reads a PDF file, encodes it into base64 and then displays it on a Streamlit app using markdown.
53
+
54
+ Parameters:
55
+ - file (str): The file path to the PDF file that needs to be displayed.
56
+
57
+ Returns:
58
+ - None
59
+ """
60
+
61
+ # Opening file from file path
62
+ with open(file, "rb") as f:
63
+ # Encoding file content to base64
64
+ base64_pdf = base64.b64encode(f.read()).decode('utf-8')
65
+
66
+ # Embedding PDF in HTML with specific size and type
67
+ pdf_display = F'<embed src="data:application/pdf;base64,{base64_pdf}" width="700" height="1000" type="application/pdf">'
68
+
69
+ # Displaying File in Streamlit app by allowing unsafe HTML
70
+ st.markdown(pdf_display, unsafe_allow_html=True)
71
+
72
+
73
+
74
  client = OpenAI(api_key=os.environ["OPENAI_API_KEY"])
75
 
76