Update src/streamlit_app.py

src/streamlit_app.py

@@ -167,14 +167,13 @@ def cached_load_embeddings_and_index():
 
 @st.cache_resource(ttl=3600)
 def init_groq_llm():
-    # Проверка на наличие ключа Groq
-    if "GROQ_API_KEY" not in st.secrets:
-        st.error("Ошибка: API-ключ Groq не найден в secrets.toml.")
-        return None
-    
-    os.environ["GROQ_API_KEY"] = st.secrets["GROQ_API_KEY"]
     try:
+        groq_api_key = st.secrets["GROQ_API_KEY"]
+        os.environ["GROQ_API_KEY"] = groq_api_key
         return ChatGroq(model="deepseek-r1-distill-llama-70b", temperature=0, max_tokens=2000)
+    except KeyError:
+        st.error("Ошибка: API-ключ Groq не найден в secrets.toml. Пожалуйста, добавьте его.")
+        return None
     except Exception as e:
         st.error(f"Ошибка инициализации Groq: {e}")
         return None
@@ -239,7 +238,7 @@ def semantic_search(query, embedder, index, df, genre=None, year=None, country=N
     if keyword_pattern:
         res['has_keyword'] = res.apply(
             lambda row: bool(re.search(keyword_pattern, str(row['tvshow_title']).lower())) or
-                        bool(re.search(keyword_pattern, str(row['description']).lower())),
+                         bool(re.search(keyword_pattern, str(row['description']).lower())),
             axis=1
         )
     else:
@@ -307,6 +306,12 @@ def main():
     st.set_page_config(page_title="Поиск фильмов и сериалов + Groq AI", layout="wide")
     st.title("📽️ Семантический поиск фильмов и сериалов с AI")
 
+    # Убедимся, что st.secrets доступен перед инициализацией LLM
+    try:
+        groq_api_key_exists = "GROQ_API_KEY" in st.secrets
+    except Exception:
+        groq_api_key_exists = False
+    
     if "df" not in st.session_state:
         st.session_state.df = cached_load_data(CSV_PATH)
     if "embedder" not in st.session_state:
@@ -314,8 +319,11 @@ def main():
     if "embeddings_index" not in st.session_state:
         with st.spinner("Загрузка эмбеддингов и индекса..."):
             st.session_state.embeddings, st.session_state.index = cached_load_embeddings_and_index()
-    if "llm" not in st.session_state:
+    if "llm" not in st.session_state and groq_api_key_exists:
         st.session_state.llm = init_groq_llm()
+    elif "llm" not in st.session_state:
+        st.session_state.llm = None
+        st.error("API-ключ Groq не настроен. Функции AI будут отключены.")
 
     if 'last_query' not in st.session_state: st.session_state.last_query = ""
     if 'results' not in st.session_state: st.session_state.results = pd.DataFrame()
@@ -437,13 +445,15 @@ def main():
     if st.session_state.llm and not st.session_state.results.empty:
         if st.button("AI: почему эти сериалы подходят и что ещё посмотреть", key="ai_button"):
             st.session_state.ai_clicked = True
+    elif not st.session_state.llm and not st.session_state.results.empty:
+        st.warning("Чтобы получить рекомендации от AI, необходимо настроить API-ключ Groq.")
     
     with ai_response_container:
-        if st.session_state.get("ai_clicked") and st.session_state.get("last_query"):
+        if st.session_state.get("ai_clicked") and st.session_state.get("last_query") and st.session_state.llm:
             st.markdown("### Рекомендации AI:")
             with st.spinner("Генерация ответа AI..."):
                 rag = generate_rag_response(st.session_state.last_query, st.session_state.results, llm)
-                st.write(rag)
+            st.write(rag)
 
     st.sidebar.markdown("---")
     st.sidebar.markdown("## Информация")