File size: 1,853 Bytes
77a3002
 
7adf0ef
77a3002
7adf0ef
705e588
77a3002
 
 
 
 
705e588
 
 
 
 
 
 
 
 
 
 
77a3002
 
7adf0ef
 
 
 
77a3002
 
 
7adf0ef
77a3002
705e588
77a3002
 
705e588
7adf0ef
77a3002
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
import gradio as gr

# Gradio application setup
def create_demo():

    with gr.Blocks(title="RAG Chatbot Q&A", theme="Soft") as demo:
        with gr.Column():
            with gr.Row():
                chat_history = gr.Chatbot(value=[], elem_id='chatbot', height=680)
                show_img = gr.Image(label='Overview', height=680)

        with gr.Column():
            with gr.Row():
                slider_chunk_size = gr.Slider(minimum=256, maximum=1024, value=256, label="Chunk Size", elem_id='slider1')
            with gr.Row():
                slider_overlap_percentage = gr.Slider(minimum=256, maximum=1024, value=256, label="Chunk Overlap Percentage", elem_id='slider2')
        with gr.Column():
            with gr.Row():
                slider_temp = gr.Slider(minimum=256, maximum=1024, value=256, label="Model Temperature", elem_id='slider3')
            with gr.Row():
                slider_k = gr.Slider(minimum=256, maximum=1024, value=256, label="Max Chunks in Context", elem_id='slider2')

        with gr.Row():
            with gr.Column(scale=0.60):
                text_input = gr.Textbox(
                    show_label=False,
                    placeholder="Type here to ask your PDF",
                container=False)
            with gr.Column(scale=0.20):
                submit_button = gr.Button('Send')
            with gr.Column(scale=0.20):
                uploaded_pdf = gr.UploadButton("📁 Upload PDF", file_types=[".pdf"], elem_id='upload_pdf')

    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, slider_chunk_size,slider_overlap_percentage,slider_temp,slider_k = create_demo()
    demo.queue()
    demo.launch()