functionality test

app.py

@@ -32,16 +32,27 @@ def load_pdf_content(pdf):
 
 
 def main():
-    st.title("GGUF model loading test")
+    st.title("BetterZila RAG Enabled LLM")
     
-    # 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)
+    with st.spinner("Loading model..."):
+        prompter = register_gguf_model()
+    st.success("Model loaded!")
+
+    with st.spinner("Loading PDF content from the assignment URL..."):
+        url = "https://pgcag.files.wordpress.com/2010/01/48lawsofpower.pdf"
+        pdf = load_pdf_from_url(url)
+        content = load_pdf_content(pdf)
+    st.success("PDF content loaded!")
+    
+    queries = ['Can you give me an example from history where the enemy was crushed totally from the book?', "What's the point of making myself less accessible?", "Can you tell me the story of Queen Elizabeth I from this 48 laws of power book?"]
+    
+    for query in queries:
+        st.subheader(f"Query: {query}")
+        with st.spinner("Generating response..."):
+            resp = prompter.prompt_main(query, context=content)
+            response = resp['llm_response']
+        st.success("Response generated!")
+        st.write(response)
         
 if __name__ == "__main__":
     main()