Update app.py

app.py

@@ -5,7 +5,7 @@ import gradio as gr
 from huggingface_hub import InferenceClient
 import requests
 import time
-import base64  # Добавлено для декодирования base64 из GitHub API
+import base64  # Для декодирования base64 из GitHub API
 
 # -----------------------------
 # Конфигурация GitHub (для приватного репозитория)
@@ -13,55 +13,64 @@ import base64  # Добавлено для декодирования base64 и
 GITHUB_OWNER = "Tony14100"
 GITHUB_REPO = "Prompt-Tonic"
 GITHUB_REF = "main"
-GITHUB_PATH = "OzonConsultant"
+GITHUB_PATH = "OzonConsultant"  # Убедитесь, что это файл в корне репо
 
 GITHUB_API_URL = f"https://api.github.com/repos/{GITHUB_OWNER}/{GITHUB_REPO}/contents/{GITHUB_PATH}?ref={GITHUB_REF}"
 
 LOCAL_PROMPT_FILE = "prompt_cache.txt"
-LOCAL_SHA_FILE = "prompt_cache_sha.txt"  # Новый файл для хранения SHA
+LOCAL_SHA_FILE = "prompt_cache_sha.txt"  # Файл для хранения SHA
 
+# -----------------------------
+# Функция загрузки системного промпта
+# -----------------------------
 def load_system_prompt():
     github_token = os.getenv("GITHUB_TOKEN")
     if not github_token:
-        raise RuntimeError("GITHUB_TOKEN не найден. Добавь его в Secrets (export GITHUB_TOKEN=твой_токен). Для приватного репозитория токен обязателен.")
+        print("[WARNING] GITHUB_TOKEN не найден. Fallback на дефолтный промпт.")
+        return "Вы — полезный консультант по Ozon. Отвечайте на вопросы клиентов четко и профессионально."  # Дефолтный промпт
 
     headers = {
         'Authorization': f'token {github_token}',
-        'User-Agent': 'Gradio-App-Request'  # GitHub требует User-Agent
+        'User-Agent': 'Gradio-App-Request',
+        'Accept': 'application/vnd.github.v3+json'
     }
 
+    print(f"[DEBUG] Запрос к: {GITHUB_API_URL}")
+
     try:
-        # Получаем метаданные и содержимое файла через API
         response = requests.get(GITHUB_API_URL, headers=headers, timeout=10)
+        print(f"[DEBUG] Status code: {response.status_code}")
+        print(f"[DEBUG] Response headers: {dict(response.headers)[:3]}...")  # Первые 3 заголовка для отладки
+
+        if response.status_code == 404:
+            raise RuntimeError(f"404: 'OzonConsultant' не найден. Проверьте: файл существует в корне на 'main'? Это файл (не папка)? Полный путь верный?")
+        
         response.raise_for_status()
         data = response.json()
 
         current_sha = data['sha']
-        print(f"[DEBUG] Текущий SHA файла: {current_sha}")
+        print(f"[DEBUG] SHA: {current_sha[:10]}...")
 
-        # Проверяем кэш SHA
+        # Проверяем кэш по SHA
         cached_sha = None
         if os.path.exists(LOCAL_SHA_FILE):
             try:
                 with open(LOCAL_SHA_FILE, 'r') as f:
                     cached_sha = f.read().strip()
-                print(f"[DEBUG] Кэшированный SHA: {cached_sha}")
             except Exception as e:
-                print(f"Ошибка чтения SHA кэша: {e}")
+                print(f"Ошибка чтения SHA: {e}")
 
-        # Если SHA не изменился, используем кэш
         if cached_sha == current_sha and os.path.exists(LOCAL_PROMPT_FILE):
             try:
                 with open(LOCAL_PROMPT_FILE, 'r', encoding='utf-8') as f:
                     prompt = f.read().strip()
                 if prompt:
-                    print("Используется кэшированный промпт (SHA не изменился).")
+                    print("Кэш SHA совпадает — используем локальный промпт.")
                     return prompt
                 else:
                     raise ValueError("Кэш пустой.")
             except Exception as e:
                 print(f"Ошибка чтения кэша: {e}")
-                # Если ошибка, продолжаем обновление
 
         # SHA изменился или нет кэша — обновляем
         print("Обновляем промпт с GitHub (SHA изменился).")
@@ -69,7 +78,7 @@ def load_system_prompt():
         prompt = base64.b64decode(content_b64).decode('utf-8').strip()
 
         if not prompt:
-            raise ValueError("Промпт пустой. Проверь файл на GitHub.")
+            raise ValueError("Промпт пустой. Проверьте файл на GitHub.")
 
         # Сохраняем в кэш
         with open(LOCAL_PROMPT_FILE, 'w', encoding='utf-8') as f:
@@ -80,17 +89,37 @@ def load_system_prompt():
         print("Промпт обновлён и сохранён в кэш.")
         return prompt
 
-    except requests.exceptions.RequestException as e:
-        raise RuntimeError(f"Ошибка запроса к GitHub API ({GITHUB_API_URL}): {e}")
-    except (KeyError, base64.binascii.Error, UnicodeDecodeError) as e:
-        raise RuntimeError(f"Ошибка парсинга ответа GitHub: {e}")
+    except requests.exceptions.HTTPError as e:
+        status = e.response.status_code if hasattr(e.response, 'status_code') else 0
+        if status == 401:
+            raise RuntimeError(f"401: Токен неверный/истёк. Создайте новый в GitHub Settings > Developer settings.")
+        elif status == 403:
+            raise RuntimeError(f"403: Токен без доступа. Добавьте 'Contents: Read' для Prompt-Tonic.")
+        else:
+            raise RuntimeError(f"HTTP {status}: {e}. Проверьте токен/URL.")
     except Exception as e:
-        raise RuntimeError(f"Не удалось загрузить промпт с GitHub: {e}")
+        print(f"[WARNING] GitHub API failed: {e}. Fallback активирован.")
+
+        # Fallback 1: Локальный файл (загрузите OzonConsultant.txt в Space)
+        local_file = "OzonConsultant.txt"
+        if os.path.exists(local_file):
+            try:
+                with open(local_file, 'r', encoding='utf-8') as f:
+                    prompt = f.read().strip()
+                print(f"Fallback: Загружен локальный {local_file}.")
+                return prompt
+            except Exception as le:
+                print(f"Локальный fallback failed: {le}")
+
+        # Fallback 2: Дефолтный промпт
+        default_prompt = """Вы — полезный консультант по Ozon. Отвечайте на вопросы клиентов четко и профессионально."""
+        print("Fallback: Дефолтный промпт (GitHub недоступен). Обновите файл/токен.")
+        return default_prompt
 
 SYSTEM_PROMPT = load_system_prompt()
 print("Загруженный промпт:", SYSTEM_PROMPT[:200] + "..." if len(SYSTEM_PROMPT) > 200 else SYSTEM_PROMPT)
 
-MODEL_ID = "openai/gpt-oss-20b"
+MODEL_ID = "zai-org/GLM-4.5"
 MAX_TOKENS = 512
 
 # -----------------------------
@@ -119,7 +148,7 @@ def on_user_message(user_message: str, state: Dict):
     if not state["system_used"]:
         messages.insert(0, {"role": "system", "content": SYSTEM_PROMPT})
         state["system_used"] = True
-    print("[DEBUG] System prompt добавлен в messages.")
+        print("[DEBUG] System prompt добавлен в messages.")
 
     messages.append({"role": "user", "content": user_message})
     assistant_reply = ""
@@ -134,7 +163,7 @@ def on_user_message(user_message: str, state: Dict):
             stream=False,
         )
         assistant_reply = response.choices[0].message["content"].strip()
-        print(f"[DEBUG] Ответ модели: {assistant_reply}")
+        print(f"[DEBUG] Ответ модели: {assistant_reply[:50]}...")
     except Exception as e:
         print(f"[DEBUG] Ошибка модели: {e}")
         assistant_reply = mock_predict(user_message)
@@ -186,11 +215,12 @@ if __name__ == "__main__":
     except Exception as e:
         print(f"[WARNING] Проблема с токеном: {e}. Установи HF_TOKEN!")
 
-    # Проверяем GITHUB_TOKEN при запуске
+    # Проверяем GITHUB_TOKEN при запуске (но не краш, если fallback)
     try:
-        load_system_prompt()  # Уже вызвано выше, но для проверки
+        # load_system_prompt() уже вызвано выше
+        pass
     except Exception as e:
-        print(f"[WARNING] Проблема с GITHUB_TOKEN: {e}. Установи GITHUB_TOKEN для приватного репозитория!")
+        print(f"[WARNING] Проблема с GITHUB_TOKEN: {e}. Fallback активен!")
 
     app.queue(max_size=5)
     app.launch(server_name="0.0.0.0", server_port=7860, debug=True, share=False)