Update

Author: Lin-jun-xiang

README.md

@@ -7,6 +7,12 @@
 
 使用該工具至少須具備 `openai_api_key`，有關如何取得 key 可以前往[連結](https://platform.openai.com/)
 
+## How to develope a docGPT with streamlit?
+
+[Tutorial](./tutorial/)
+
+
+
 ## Introduction
 
 * `docGPT` 是基於 langchain 與 streamlit 開發的

app.py

@@ -20,16 +20,23 @@
 SERPAPI_API_KEY = ''
 agent_ = None
 
-st.set_page_config(page_title="DocGPT")
-icon, title = st.columns([3, 20])
-with icon:
-    st.image('./img/chatbot.png')
-with title:
-    st.title('PDF Chatbot')
 st.session_state.openai_api_key = None
 st.session_state.serpapi_api_key = None
 
 
+def theme():
+    st.set_page_config(page_title="DocGPT")
+
+    icon, title = st.columns([3, 20])
+    with icon:
+        st.image('./img/chatbot.png')
+    with title:
+        st.title('PDF Chatbot')
+
+
+theme()
+
+
 def load_api_key() -> None:
     with st.sidebar:
 

docGPT/__init__.py

@@ -1,5 +1 @@
 from .docGPT import DocGPT
-from .response_processor import (
-    JSONProcessor,
-    slice_list,
-)

model/__init__.py

@@ -1,4 +1,3 @@
 from .data_connection import (
-    JSONWriter,
     PDFLoader
 )

model/data_connection.py

@@ -35,28 +35,3 @@ def split_documents(
         )
 
         return splitter.split_documents(document)
-
-
-class JSONWriter:
-    def __init__(self, file_name: str) -> None:
-        self.file_name = file_name
-    
-    def writer(self, data: list) -> None:
-        if os.path.exists(self.file_name):
-            with open(self.file_name) as file:
-                listObj = json.load(file)
-
-            listObj.extend(data)
-            data = listObj
-
-        with open(self.file_name, 'w') as file:
-            json.dump(
-                data,
-                file, 
-                indent=4
-            )
-
-
-class LoggerHandle:
-    def qa_chain(self, chain_type: str, retriever, llm):
-        pass

requirements.txt

@@ -0,0 +1,4 @@
+langchain==0.0.222
+openai==0.27.8
+streamlit==1.22.0
+pymupdf