Update app.py

app.py

@@ -1,18 +1,14 @@
+import pdfminer
+from pdfminer.high_level import extract_pages
 from transformers import pipeline
 
 import streamlit as st
 
 
 def get_openai_response(question):
-    # llm = OpenAI(
-    #     openai_api_key=os.getenv("OPEN_API_KEY"),
-    #     model_name="gpt-3.5-turbo-instruct",
-    #     temperature=0.6,
-    # )
     response = pipeline("summarization", model="stevhliu/my_awesome_billsum_model")
-    # response = llm(question)
     return response
-    # modify with chain and other stuff
+
 
 
 ## streamlit app
@@ -23,9 +19,13 @@ st.write("UPDATE: This app uses the 'gpt-3.5-turbo-instruct' model through Langc
 
 
 # input = st.text_input("Enter your query: ", key=input)
-uploaded_file = st.file_uploader('Choose your .pdf file', type="pdf")
+st.write(pdfminer.__version__)  
+
+uploaded_file = st.file_uploader("Choose a file", "pdf")
 if uploaded_file is not None:
-    df = extract_data(uploaded_file)
+    for page_layout in extract_pages(uploaded_file):
+        for element in page_layout:
+            st.write(element)
 response = get_openai_response(df)