updated app.py

app.py

@@ -9,11 +9,11 @@ def respond(message, chat_history, collection_name):
     return "", chat_history
 
 
-def embed_and_store(password, collection_name, file_type, file_fields, context):
+def embed_and_store(password, collection_name, file_type, file_fields, context,page_start):
     if password == config['PASSWORD_DB']:
         if str(file_type)== 'string':
             file_fields = context
-        vector_db = VectorDataBase(file_fields, collection_name, file_type)
+        vector_db = VectorDataBase(file_fields, collection_name, file_type, page_start=page_start)
         vector_db.embedding_with_loop()
         return "", ""
     else:
@@ -51,13 +51,14 @@ with gr.Blocks() as demo:
         with gr.Row():
             password = gr.Textbox(label='Enter the Password')
             collection_name = gr.Textbox(label='Collection Name')
+            page_start = gr.Textbox(label='Page Start')
         file_type = gr.Dropdown(['PDF', 'TEXT', 'STRING'], label='Select File Type',
                                value = 'PDF')
         file_fields = gr.File(visible = True, interactive=True)
         context = gr.Textbox(label="Enter the Context", visible = False)
         btn = gr.Button("Submit")
 
-        btn.click(embed_and_store, inputs=[password, collection_name, file_type, file_fields, context], outputs=[file_fields, context])
+        btn.click(embed_and_store, inputs=[password, collection_name, file_type, file_fields, context,page_start], outputs=[file_fields, context])
         file_type.change(update_interface, inputs=[file_type], outputs=[context, file_fields])
     gr.Markdown("""
     <div align='center'>It could be helpful for making RAG applications</div>