Update app.py

app.py

@@ -6,10 +6,22 @@ from transformers import pipeline
 from kaggle.api.kaggle_api_extended import KaggleApi
 
 # === Подготовка банковского набора данных через Kaggle ===
-# Будет скачан dataset PromptCloudHQ/banking-chatbot-dataset, содержащий примеры вопросов и ответов для банковского чат-бота.
+# Пишем файл kaggle.json из переменных окружения, чтобы authenticate() нашёл его
+import json
+kaggle_config_dir = os.path.expanduser("~/.config/kaggle")
+os.makedirs(kaggle_config_dir, exist_ok=True)
+kaggle_json = {
+    "username": os.getenv("KAGGLE_USERNAME", ""),
+    "key": os.getenv("KAGGLE_KEY", "")
+}
+if not kaggle_json["username"] or not kaggle_json["key"]:
+    raise ValueError("Не найдены переменные окружения KAGGLE_USERNAME и KAGGLE_KEY")
+with open(os.path.join(kaggle_config_dir, "kaggle.json"), "w") as f:
+    json.dump(kaggle_json, f)
+
+# Скачиваем при первом запуске
 DATA_DIR = './data'
 json_file = None
-# Скачиваем при первом запуске
 if not os.path.exists(DATA_DIR):
     os.makedirs(DATA_DIR)
     api = KaggleApi()
@@ -21,7 +33,6 @@ if not os.path.exists(DATA_DIR):
             json_file = os.path.join(DATA_DIR, fname)
             break
 else:
-    # Если папка есть — ищем файл
     for fname in os.listdir(DATA_DIR):
         if fname.endswith('.json'):
             json_file = os.path.join(DATA_DIR, fname)
@@ -30,22 +41,6 @@ else:
 if json_file is None:
     raise FileNotFoundError('Не удалось найти JSON-файл с банковскими данными в ./data')
 
-# Загружаем JSON с примерами
-with open(json_file, 'r', encoding='utf-8') as f:
-    kb = json.load(f)
-# Структура: {"intents": [ ... ]}
-intents = kb.get('intents')
-if intents is None:
-    raise ValueError('Ожидался ключ "intents" в JSON-файле датасета')
-
-# Собираем два few-shot примера
-examples = []
-for intent in intents[:2]:
-    patterns = intent.get('patterns', [])
-    responses = intent.get('responses', [])
-    ex = f"Паттерны: {', '.join(patterns)}\nОтветы: {', '.join(responses)}"
-    examples.append(ex)
-
 # === Инициализация трёх бесплатных русскоязычных моделей (GPT-2 based) ===
 models = {
     'ruDialoGPT-small': pipeline('text-generation', model='t-bank-ai/ruDialoGPT-small', tokenizer='t-bank-ai/ruDialoGPT-small', device=-1),
@@ -108,3 +103,4 @@ with gr.Blocks() as demo:
     t3 = gr.Textbox(label='ruGPT3-small Время')
     btn.click(format_outputs, inputs=[txt], outputs=[out1, t1, out2, t2, out3, t3])
     demo.launch()
+