Update app.py

app.py

@@ -307,21 +307,67 @@ def demo():
             db_progress = gr.Textbox(label="Vector Database Initialization Status", value="None", interactive=False)
             generate_db_btn = gr.Button("Generate Vector Database")
 
-        with gr.Tab("Step 3 - Initialize QA chain"):
-            with gr.Row():
-                llm_btn = gr.Radio(list_llm_simple, \
-                    label="LLM models", value = list_llm_simple[0], type="index", info="Choose your LLM model")
-            with gr.Accordion("Advanced options - LLM model", open=False):
-                with gr.Row():
-                    slider_temperature = gr.Slider(minimum = 0.01, maximum = 1.0, value=0.7, step=0.1, label="Temperature", info="Model temperature", interactive=True)
-                with gr.Row():
-                    slider_maxtokens = gr.Slider(minimum = 224, maximum = 4096, value=1024, step=32, label="Max Tokens", info="Model max tokens", interactive=True)
-                with gr.Row():
-                    slider_topk = gr.Slider(minimum = 1, maximum = 10, value=3, step=1, label="top-k samples", info="Model top-k samples", interactive=True)
-            with gr.Row():
-                llm_progress = gr.Textbox(value="None",label="QA chain initialization")
-            with gr.Row():
-                qachain_btn = gr.Button("Initialize Question Answering chain")
+        # with gr.Tab("Step 3 - Initialize QA chain"):
+        #     with gr.Row():
+        #         llm_btn = gr.Radio(list_llm_simple, \
+        #             label="LLM models", value = list_llm_simple[0], type="index", info="Choose your LLM model")
+        #     with gr.Accordion("Advanced options - LLM model", open=False):
+        #         with gr.Row():
+        #             slider_temperature = gr.Slider(minimum = 0.01, maximum = 1.0, value=0.7, step=0.1, label="Temperature", info="Model temperature", interactive=True)
+        #         with gr.Row():
+        #             slider_maxtokens = gr.Slider(minimum = 224, maximum = 4096, value=1024, step=32, label="Max Tokens", info="Model max tokens", interactive=True)
+        #         with gr.Row():
+        #             slider_topk = gr.Slider(minimum = 1, maximum = 10, value=3, step=1, label="top-k samples", info="Model top-k samples", interactive=True)
+        #     with gr.Row():
+        #         llm_progress = gr.Textbox(value="None",label="QA chain initialization")
+        #     with gr.Row():
+        #         qachain_btn = gr.Button("Initialize Question Answering chain")
+
+        with gr.Tab("Step 3 - Initialize QA Chain"):
+    with gr.Row():
+        llm_btn = gr.Radio(
+            ["Mistral 7B"],  # Only Mistral 7B is listed
+            label="LLM Model",
+            value="Mistral 7B",  # Default selection is Mistral 7B
+            type="index",
+            info="Using Mistral 7B LLM model"
+        )
+    with gr.Accordion("Advanced Options - LLM Model", open=False):
+        with gr.Row():
+            slider_temperature = gr.Slider(
+                minimum=0.01, 
+                maximum=1.0, 
+                value=0.7, 
+                step=0.1, 
+                label="Temperature", 
+                info="Adjusts the creativity of the model", 
+                interactive=True
+            )
+        with gr.Row():
+            slider_maxtokens = gr.Slider(
+                minimum=224, 
+                maximum=4096, 
+                value=1024, 
+                step=32, 
+                label="Max Tokens", 
+                info="Sets the maximum token limit", 
+                interactive=True
+            )
+        with gr.Row():
+            slider_topk = gr.Slider(
+                minimum=1, 
+                maximum=10, 
+                value=3, 
+                step=1, 
+                label="Top-K Samples", 
+                info="Selects the number of top-k samples", 
+                interactive=True
+            )
+    with gr.Row():
+        llm_progress = gr.Textbox(value="None", label="QA Chain Initialization Status", interactive=False)
+    with gr.Row():
+        qachain_btn = gr.Button("Initialize Question Answering Chain")
+
 
         with gr.Tab("Step 4: Chatbot"):
             chatbot = gr.Chatbot(label="Chat with your PDF", height=300)