Update app.py

app.py

@@ -136,6 +136,8 @@ def answer_with_llm(query, re_ranked_res, llm_type, llm_answer_type):
 
 def check_request_validity(func):
     def wrapper(
+        chatgpt_key,
+        gigachat_key,
         query,
         llm_type,
         llm_answer_type,
@@ -148,7 +150,7 @@ def check_request_validity(func):
             return 'Невалидный запрос', '', ''
             
         if llm_type == 'Без LLM' or not use_llm_for_request_validation:
-            return func(query, llm_type, llm_answer_type, use_llm_for_retriever)
+            return func(chatgpt_key, gigachat_key, query, llm_type, llm_answer_type, use_llm_for_retriever)
         
         llm_chain = LLMChain(prompt=validity_prompt, llm=llms[llm_type])
 
@@ -203,8 +205,8 @@ def fn(
 demo = gr.Interface(
     fn=fn,
     inputs=[
-        gr.Textbox(lines=3, label='Ключ Open AI API', placeholder='Введите ключ'),
-        gr.Textbox(lines=3, label='Ключ GigaChat API', placeholder='Введите ключ'),
+        gr.Textbox(lines=1, label='Ключ Open AI API', placeholder='Введите ключ'),
+        gr.Textbox(lines=1, label='Ключ GigaChat API', placeholder='Введите ключ'),
         gr.Textbox(lines=3, label='Запрос', placeholder='Введите запрос'),
         gr.Dropdown(label='Тип LLM', choices=['ChatGPT', 'GigaChat', 'Без LLM'], value='ChatGPT'),
         gr.Dropdown(label='Тип итогового ответа LLM', choices=['Только цифры штрафа', 'Развернутый ответ'], value='Только цифры штрафа'),