Update app.py

app.py

@@ -403,7 +403,7 @@ def create_vector_db_and_annoy_index(pdf_path, vector_db_path, annoy_index_path)
     print("Vector database and Annoy index creation completed.")
 
 
-def chatbot_interface(file, user_query, response_style, selected_retrieval_methods, selected_reranking_methods, chunk_size, overlap):
+def chatbot_interface(file, user_query, response_style):
     vector_db_path = "vector_db.pkl"
     annoy_index_path = "vector_index.ann"
     
@@ -411,8 +411,8 @@ def chatbot_interface(file, user_query, response_style, selected_retrieval_metho
 
     chatbot = MistralRAGChatbot(vector_db_path, annoy_index_path)
 
-    selected_retrieval_methods_list = [method.strip() for method in selected_retrieval_methods if method.strip()]
-    selected_reranking_methods_list = [method.strip() for method in selected_reranking_methods.split(',') if method.strip()]
+    selected_retrieval_methods_list = ['annoy', 'bm25']
+    selected_reranking_methods_list = ["advanced_fusion"]
 
     response, retrieved_docs, source_info = asyncio.run(chatbot.generate_response_with_rag(
         user_query=user_query,
@@ -440,11 +440,9 @@ iface = gr.Interface(
     inputs=[
         gr.File(label="Upload a PDF"), 
         gr.Textbox(lines=5, label="User Query"),
-        gr.Dropdown(["Detailed", "Concise", "Creative", "Technical"], label="Response Style"),
-        gr.Dropdown(["annoy", "tfidf", "bm25", "euclidean", "jaccard"], label="Retrieval Methods", multiselect=True),
-        gr.Dropdown(["advanced_fusion", "reciprocal_rank_fusion", "weighted_score_fusion", "semantic_similarity"], label="Reranking Methods"),
-        gr.Slider(minimum=1024, maximum=2048, step=128, value=2048, label="Chunk Size"),
-        gr.Slider(minimum=100, maximum=300, step=100, value=200, label="Overlap")
+        gr.Dropdown(["Detailed", "Concise", "Creative", "Technical"], label="Retreival Style"),
+        gr.Markdown("#Retreival Method"),
+        gr.Mardown("##advanced_fusion")
     ],
     outputs=gr.Textbox(label="RAG Response"),
     title="Chance RAG"