Update src/interface.py

src/interface.py

@@ -1,20 +1,14 @@
-import csv
 import gradio as gr
 
 # Gradio application setup
 def create_demo():
-    with gr.Blocks(title="LLAMA 3 Rag Chat pdf", theme="Monochrome") as demo:
+    with gr.Blocks(title="LLAMA 3 Rag on Fly", theme="Monochrome") as demo:
         # App Description
         gr.Markdown(
             """
-            ## LLAMA 3 Rag chat pdf
-
+            ## LLAMA 3 Rag on Fly App
             This application allows you to experiment with LLAMA 3 8B Instruct model for RAG.             
-            You can adjust various parameters to control the model's output. 
-
-            Original https://huggingface.co/spaces/ModularityAI/LLama3Rag
-
-            experimented on making RAG for long pdf documents
+            You can adjust various parameters to control the model's output.             
             """
         )
 
@@ -29,7 +23,7 @@ def create_demo():
             with gr.Column(scale=0.05):
                 with gr.Row():
                     slider_chunk_size = gr.Slider(
-                        minimum=256, maximum=32768, value=1024, label="Chunk Size", elem_id='slider1'
+                        minimum=256, maximum=1024, value=256, label="Chunk Size", elem_id='slider1'
                     )
                 with gr.Row():
                     slider_overlap_percentage = gr.Slider(
@@ -41,7 +35,7 @@ def create_demo():
                     )
                 with gr.Row():
                     slider_k = gr.Slider(
-                        minimum=1, step=1,maximum=20, value=2, label="Max Chunks in Context", elem_id='slider2'
+                        minimum=1, step=1,maximum=5, value=2, label="Max Chunks in Context", elem_id='slider2'
                     )
 
         # Input and Submit Button Row
@@ -56,15 +50,10 @@ def create_demo():
                 submit_button = gr.Button('Send')
             with gr.Column(scale=0.20):
                 uploaded_pdf = gr.UploadButton("📁 Upload PDF", file_types=[".pdf"], elem_id='upload_pdf')
-            #NEW
-            with gr.Column(scale=0.20):
-                download_chat_history = gr.DownloadButton('Download chat History')
-            #NEW
-
-    return demo, chat_history, show_img, text_input, submit_button, uploaded_pdf, download_chat_history, slider_chunk_size,slider_overlap_percentage,slider_temp,slider_k
 
+    return demo, chat_history, show_img, text_input, submit_button, uploaded_pdf, slider_chunk_size,slider_overlap_percentage,slider_temp,slider_k
 
 if __name__ == '__main__':
-    demo, chatbot, show_img, text_input, submit_button, uploaded_pdf, download_chat_history, slider_chunk_size,slider_overlap_percentage,slider_temp,slider_k = create_demo()
+    demo, chatbot, show_img, text_input, submit_button, uploaded_pdf, slider_chunk_size,slider_overlap_percentage,slider_temp,slider_k = create_demo()
     demo.queue()
     demo.launch()