pdf content loading test

app.py

@@ -1,5 +1,8 @@
 import streamlit as st
 from llmware.prompts import Prompt
+import requests
+import io
+import PyPDF2
 
 def register_gguf_model():
 
@@ -15,13 +18,30 @@ def register_gguf_model():
 
     return prompter
 
+def load_pdf_from_url(url):
+    r = requests.get(url)
+    f = io.BytesIO(r.content)
+    file = PyPDF2.PdfReader(f)
+    return file
+
+def load_pdf_content(pdf):
+    content = ""
+    for page in pdf.pages:
+        content += page.extract_text()
+    return content
+
+
 def main():
     st.title("GGUF model loading test")
     
-    with st.spinner("Loading model..."):
-        prompter = register_gguf_model()
-    if prompter:
-        st.success("Model loaded!")
+    # with st.spinner("Loading model..."):
+    #     prompter = register_gguf_model()
+    # if prompter:
+    #     st.success("Model loaded!")
+    url = "https://pgcag.files.wordpress.com/2010/01/48lawsofpower.pdf"
+    pdf = load_pdf_from_url(url)
+    content = load_pdf_content(pdf)
+    st.write(content)
         
 if __name__ == "__main__":
     main()

requirements.txt

@@ -1,2 +1,5 @@
 llmware
-streamlit
+streamlit
+requests
+io
+PyPDF2