init

requirements.txt

@@ -1,3 +1,97 @@
 altair
-pandas
-streamlit
+altair==5.5.0
+annotated-types==0.7.0
+anyio==4.9.0
+appnope==0.1.4
+asttokens==3.0.0
+async-timeout==4.0.3
+attrs==25.3.0
+blinker==1.9.0
+cachetools==6.1.0
+certifi==2025.6.15
+charset-normalizer==3.4.2
+click==8.1.8
+comm==0.2.2
+debugpy==1.8.14
+decorator==5.2.1
+dotenv==0.9.9
+exceptiongroup==1.3.0
+executing==2.2.0
+gigachat==0.1.40
+gitdb==4.0.12
+GitPython==3.1.44
+h11==0.16.0
+httpcore==1.0.9
+httpx==0.28.1
+idna==3.10
+importlib_metadata==8.7.0
+ipykernel==6.29.5
+ipython==8.18.1
+jedi==0.19.2
+Jinja2==3.1.6
+jsonpatch==1.33
+jsonpointer==3.0.0
+jsonschema==4.24.0
+jsonschema-specifications==2025.4.1
+jupyter_client==8.6.3
+jupyter_core==5.8.1
+langchain==0.3.26
+langchain-core==0.3.66
+langchain-gigachat==0.3.11
+langchain-text-splitters==0.3.8
+langgraph==0.5.0
+langgraph-checkpoint==2.1.0
+langgraph-prebuilt==0.5.1
+langgraph-sdk==0.1.72
+langsmith==0.4.4
+MarkupSafe==3.0.2
+matplotlib-inline==0.1.7
+narwhals==1.44.0
+nest-asyncio==1.6.0
+numpy==2.0.2
+orjson==3.10.18
+ormsgpack==1.10.0
+packaging==24.2
+pandas==2.3.0
+parso==0.8.4
+pexpect==4.9.0
+pillow==11.2.1
+platformdirs==4.3.8
+prompt_toolkit==3.0.51
+protobuf==6.31.1
+psutil==7.0.0
+ptyprocess==0.7.0
+pure_eval==0.2.3
+pyarrow==20.0.0
+pydantic==2.11.7
+pydantic_core==2.33.2
+pydeck==0.9.1
+Pygments==2.19.2
+python-dateutil==2.9.0.post0
+python-dotenv==1.1.1
+pytz==2025.2
+PyYAML==6.0.2
+pyzmq==27.0.0
+referencing==0.36.2
+requests==2.32.4
+requests-toolbelt==1.0.0
+rpds-py==0.25.1
+six==1.17.0
+smmap==5.0.2
+sniffio==1.3.1
+SQLAlchemy==2.0.41
+stack-data==0.6.3
+streamlit==1.46.1
+tenacity==9.1.2
+toml==0.10.2
+tornado==6.5.1
+traitlets==5.14.3
+types-requests==2.32.4.20250611
+typing-inspection==0.4.1
+typing_extensions==4.14.0
+tzdata==2025.2
+urllib3==2.5.0
+wcwidth==0.2.13
+xxhash==3.5.0
+zipp==3.23.0
+zstandard==0.23.0

src/streamlit_app.py

@@ -1,40 +1,101 @@
-import altair as alt
-import numpy as np
-import pandas as pd
 import streamlit as st
+from langchain_core.messages import HumanMessage, AIMessage
+from services.get_classification import get_graph_class
+from datetime import datetime
 
-"""
-# Welcome to Streamlit!
-
-Edit `/streamlit_app.py` to customize this app to your heart's desire :heart:.
-If you have any questions, checkout our [documentation](https://docs.streamlit.io) and [community
-forums](https://discuss.streamlit.io).
-
-In the meantime, below is an example of what you can do with just a few lines of code:
-"""
-
-num_points = st.slider("Number of points in spiral", 1, 10000, 1100)
-num_turns = st.slider("Number of turns in spiral", 1, 300, 31)
-
-indices = np.linspace(0, 1, num_points)
-theta = 2 * np.pi * num_turns * indices
-radius = indices
-
-x = radius * np.cos(theta)
-y = radius * np.sin(theta)
-
-df = pd.DataFrame({
-    "x": x,
-    "y": y,
-    "idx": indices,
-    "rand": np.random.randn(num_points),
-})
-
-st.altair_chart(alt.Chart(df, height=700, width=700)
-    .mark_point(filled=True)
-    .encode(
-        x=alt.X("x", axis=None),
-        y=alt.Y("y", axis=None),
-        color=alt.Color("idx", legend=None, scale=alt.Scale()),
-        size=alt.Size("rand", legend=None, scale=alt.Scale(range=[1, 150])),
-    ))
+
+def message_to_dict(messages):
+    result = []
+    for message in messages:
+        if isinstance(message, HumanMessage) or isinstance(message, AIMessage):
+            print("message:", message.content)
+        if message.content == "" or message.content is None:
+            continue
+        if isinstance(message, HumanMessage):
+            result.append({"role": "user", "content": message.content})
+        elif isinstance(message, AIMessage):
+            result.append({"role": "assistant", "content": message.content})
+    print("-" * 100)
+    return result
+
+
+def find_last_bot_message(messages):
+    """Находит последнее сообщение бота"""
+    for message in messages[::-1]:
+        if isinstance(message, AIMessage) and len(message.content) > 0:
+            return message.content
+    return None
+
+
+def display_chat_messages():
+    """Отображает историю сообщений в чате"""
+    for message in st.session_state.messages:
+        with st.chat_message(message["role"]):
+            st.markdown(message["content"])
+
+
+def save_broken_case():
+    messages_dict = st.session_state.messages
+    result_str = ""
+    for elem in messages_dict:
+        role = elem["role"]
+        content = elem["content"]
+        result_str += f"{role}: {content}\n"
+    
+    current_datetime = datetime.now()
+    formatted_datetime = current_datetime.strftime("%Y-%m-%d %H:%M:%S")
+
+    with open("/Users/admin/my_documents/retrieval_part/services/broken_cases.txt", "a") as file:
+        file.write(f"{formatted_datetime}\n" + result_str + "\n" + "-" * 50 + "\n\n")
+
+
+def handle_user_input():
+    """Обрабатывает ввод пользователя и генерирует ответ бота"""
+    if prompt := st.chat_input("Введите ваш вопрос"):
+        st.session_state.messages.append({"role": "user", "content": prompt})
+        if prompt.lower().startswith("log"):
+            save_broken_case()
+            st.session_state.messages = []
+            display_chat_messages()
+        else:
+            with st.chat_message("user"):
+                st.markdown(prompt)
+            
+            if "bot" not in st.session_state:
+                st.session_state.bot = get_graph_class(prompt)
+            st.session_state.bot.invoke(prompt)
+            
+            # Извлекаем последнее сообщение бота
+            last_bot_message = find_last_bot_message(st.session_state.bot.messages)
+            st.session_state.messages.append(
+                {"role": "assistant", "content": last_bot_message}
+            )
+            with st.chat_message("assistant"):
+                st.markdown(last_bot_message)
+
+
+def clear_chat():
+    """Очищает чат и пересоздает бота"""
+    st.session_state.messages = []
+    del st.session_state.bot
+
+
+def main():
+    """Основная функция приложения"""
+    # Заголовок приложения
+    st.title("Чат-бот технической поддержки OpenVPN")
+
+    # Кнопка очистки чата
+    if st.button("Clear"):
+        clear_chat()
+
+    if "messages" not in st.session_state:
+        st.session_state.messages = []
+
+    # Отображение чата и обработка ввода
+    display_chat_messages()
+    handle_user_input()
+
+
+if __name__ == "__main__":
+    main()