custom slider added

requirements.txt

@@ -1,13 +1,14 @@
-PyMuPDF==1.23.17
-gradio==4.11.0
-Pillow==10.1.0
-torch==2.1.1        
-transformers==4.35.2
-PyYAML==6.0.1
-chromadb==0.4.15
-pypdf==4.0.0
-Jinja2==3.1.3
-accelerate==0.26.1
-sentence-transformers==2.2.2
+PyMuPDF
+gradio
+Pillow
+torch
+transformers
+PyYAML
+pypdf
+Jinja2
+accelerate
+sentence-transformers
 langchain
-langchain-text-splitters
+langchain-text-splitters
+langchain_community
+langchain_core

src/app.py

@@ -2,10 +2,10 @@ from interface import create_demo
 from pdfchatbot import PDFChatBot
 
 # Create Gradio interface
-demo, chat_history, show_img, txt, submit_button, uploaded_pdf, _ = create_demo()
+demo, chat_history, show_img, txt, submit_button, uploaded_pdf, slider1 = create_demo()
 
 # Create PDFChatBot instance
-pdf_chatbot = PDFChatBot()
+pdf_chatbot = PDFChatBot(chunk_size_slider=slider1)
 pdf_chatbot.load_model()
 pdf_chatbot.load_tokenizer()
 

src/pdfchatbot.py

@@ -16,7 +16,7 @@ from langchain_text_splitters import CharacterTextSplitter
 
 
 class PDFChatBot:
-    def __init__(self, config_path="config.yaml"):
+    def __init__(self, config_path="config.yaml",chunk_size_slider=None):
         """
         Initialize the PDFChatBot instance.
 
@@ -37,7 +37,7 @@ class PDFChatBot:
         self.pipeline = None
         self.chain = None
         self.chunk_size = None
-        #self.chunk_size_slider = chunk_size_slider
+        self.chunk_size_slider = chunk_size_slider
 
     def load_config(self, file_path):
         """
@@ -71,7 +71,7 @@ class PDFChatBot:
         if not text:
             raise gr.Error('Enter text')
         history.append((text, ''))
-        #self.chunk_size_slider.interactive = False
+        self.chunk_size_slider.interactive = False
         return history
 
     def create_prompt_template(self):
@@ -180,9 +180,9 @@ class PDFChatBot:
         self.chat_history.append((query, result["answer"]))
         for char in result['answer']:
             history[-1][-1] += char
-        return history, " "
+        return result, " "
 
-    def render_file(self, file,chunk_size=256):
+    def render_file(self, file,chunk_size):
         """
         Renders a specific page of a PDF file as an image.
 
@@ -192,8 +192,8 @@ class PDFChatBot:
         Returns:
             PIL.Image.Image: The rendered page as an image.
         """
-        #if self.chunk_size_slider.interactive == False:
-        #    self.chunk_size_slider.interactive = True
+        if self.chunk_size_slider.interactive == False:
+            self.chunk_size_slider.interactive = True
         print(chunk_size)
         doc = fitz.open(file.name)
         page = doc[self.page]