buttons

app.py

@@ -16,16 +16,20 @@ embedding_model = HuggingFaceEmbeddings(
 # vector DB 로드드
 save_path = "./my_faiss_index"
 vectorstore = FAISS.load_local(save_path, embedding_model,allow_dangerous_deserialization=True)
-retriever = vectorstore.as_retriever(search_kwargs={"k": 3})
 
 
 
-def chatbot(input_question):
-    eng = translate_ko_to_en(input_question)
 
-    basic_docs = retriever.invoke(input_question)
-    eng_docs = retriever.invoke(eng)
-    basic_docs = basic_docs + eng_docs
+def chatbot(input_question, eng_trans = True, num_ref = 3):
+    retriever = vectorstore.as_retriever(search_kwargs={"k": num_ref})
+    if eng_trans == False:
+        basic_docs = retriever.invoke(input_question)
+    else:
+        eng = translate_ko_to_en(input_question)
+
+        basic_docs = retriever.invoke(input_question)
+        eng_docs = retriever.invoke(eng)
+        basic_docs = basic_docs + eng_docs
 
     context = "\n".join([doc.page_content for doc in basic_docs])  
 
@@ -57,5 +61,9 @@ def chatbot(input_question):
 
 
 
-demo = gr.Interface(fn=chatbot, inputs="text", outputs="text")
+demo = gr.Interface(fn=chatbot, 
+                    inputs=[gr.Textbox(label="질문 입력"),  # 텍스트 입력
+                            gr.Checkbox(label="영어 번역 사용 여부", value=True),  # 체크박스 추가
+                            gr.Slider(1,5, value=3, step=1),
+    ], outputs="text")
 demo.launch()