la04 commited on
Commit
d14d249
·
verified ·
1 Parent(s): 6dedc06

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +34 -13
app.py CHANGED
@@ -80,20 +80,41 @@ def demo():
80
  vector_db = gr.State()
81
  qa_chain = gr.State()
82
  gr.Markdown("<center><h1>PDF-Chatbot mit kostenlosen Modellen</h1></center>")
83
- document = gr.Files(label="PDF-Dokument hochladen")
84
- db_btn = gr.Button("Erstelle Vektordatenbank")
85
- llm_btn = gr.Radio(["Flan-T5 Small", "Flan-T5 Base"], label="Verfügbare LLMs", value="Flan-T5 Small", type="index")
86
- slider_temperature = gr.Slider(0.01, 1.0, 0.5, label="Temperature")
87
- slider_maxtokens = gr.Slider(128, 2048, 512, label="Max Tokens")
88
- slider_topk = gr.Slider(1, 10, 3, label="Top-k")
89
- qachain_btn = gr.Button("Initialisiere QA-Chatbot")
90
- chatbot = gr.Chatbot(label="Chatbot", height=400, type="messages")
91
- msg = gr.Textbox(label="Frage stellen")
92
- submit_btn = gr.Button("Absenden")
93
 
94
- db_btn.click(initialize_database, inputs=[document], outputs=[vector_db])
95
- qachain_btn.click(initialize_LLM, inputs=[llm_btn, slider_temperature, slider_maxtokens, slider_topk, vector_db], outputs=[qa_chain])
96
- submit_btn.click(conversation, inputs=[qa_chain, msg, chatbot], outputs=[qa_chain, chatbot, chatbot])
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
97
 
98
  demo.launch(debug=True)
99
 
 
80
  vector_db = gr.State()
81
  qa_chain = gr.State()
82
  gr.Markdown("<center><h1>PDF-Chatbot mit kostenlosen Modellen</h1></center>")
 
 
 
 
 
 
 
 
 
 
83
 
84
+ with gr.Row():
85
+ with gr.Column():
86
+ document = gr.Files(label="PDF-Dokument hochladen")
87
+ db_btn = gr.Button("Erstelle Vektordatenbank")
88
+ db_status = gr.Textbox(label="Datenbankstatus", value="Nicht erstellt", interactive=False)
89
+
90
+ llm_btn = gr.Radio(["Flan-T5 Small", "Flan-T5 Base"], label="Verfügbare LLMs", value="Flan-T5 Small", type="index")
91
+ slider_temperature = gr.Slider(0.01, 1.0, 0.5, label="Temperature")
92
+ slider_maxtokens = gr.Slider(128, 2048, 512, label="Max Tokens")
93
+ slider_topk = gr.Slider(1, 10, 3, label="Top-k")
94
+ qachain_btn = gr.Button("Initialisiere QA-Chatbot")
95
+ llm_status = gr.Textbox(label="Chatbot-Status", value="Nicht initialisiert", interactive=False)
96
+
97
+ with gr.Column():
98
+ chatbot = gr.Chatbot(label="Chatbot", height=400, type="messages")
99
+ msg = gr.Textbox(label="Frage stellen")
100
+ submit_btn = gr.Button("Absenden")
101
+
102
+ # Event-Handling
103
+ db_btn.click(
104
+ initialize_database,
105
+ inputs=[document],
106
+ outputs=[vector_db, db_status]
107
+ )
108
+ qachain_btn.click(
109
+ initialize_LLM,
110
+ inputs=[llm_btn, slider_temperature, slider_maxtokens, slider_topk, vector_db],
111
+ outputs=[qa_chain, llm_status]
112
+ )
113
+ submit_btn.click(
114
+ conversation,
115
+ inputs=[qa_chain, msg, chatbot],
116
+ outputs=[qa_chain, chatbot, chatbot]
117
+ )
118
 
119
  demo.launch(debug=True)
120