new5

src/streamlit_app.py

@@ -1,101 +1,107 @@
-import streamlit as st
-from langchain_core.messages import HumanMessage, AIMessage
-from get_classification import get_graph_class
-from datetime import datetime
-
-
-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"
+# import streamlit as st
+# from langchain_core.messages import HumanMessage, AIMessage
+# from get_classification import get_graph_class
+# from datetime import datetime
+
+
+# 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)
+#     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)
+#             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)
+#             # Извлекаем последнее сообщение бота
+#             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 clear_chat():
-    """Очищает чат и пересоздает бота"""
-    st.session_state.messages = []
-    del st.session_state.bot
+# def main():
+#     """Основная функция приложения"""
+#     # Заголовок приложения
+#     st.title("Чат-бот технической поддержки OpenVPN")
 
+#     # Кнопка очистки чата
+#     if st.button("Clear"):
+#         clear_chat()
 
-def main():
-    """Основная функция приложения"""
-    # Заголовок приложения
-    st.title("Чат-бот технической поддержки OpenVPN")
+#     if "messages" not in st.session_state:
+#         st.session_state.messages = []
 
-    # Кнопка очистки чата
-    if st.button("Clear"):
-        clear_chat()
+#     # Отображение чата и обработка ввода
+#     display_chat_messages()
+#     handle_user_input()
 
-    if "messages" not in st.session_state:
-        st.session_state.messages = []
 
-    # Отображение чата и обработка ввода
-    display_chat_messages()
-    handle_user_input()
+# if __name__ == "__main__":
+#     main()
 
 
-if __name__ == "__main__":
-    main()
+import streamlit as st
+
+st.title("Моё приложение")
+st.write("Привет, мир!")