Spaces:
Sleeping
Sleeping
File size: 4,682 Bytes
c5fe829 7d80464 dafe229 c5fe829 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 | import streamlit as st
import os
from langchain_core.messages import AIMessage, HumanMessage
from agent import create_agent, create_github_issue
from dotenv import load_dotenv
from ingest import main as run_ingestion
load_dotenv()
st.set_page_config(page_title="Customer Support AI", page_icon="🤖")
# To'g'ri yo'llarni belgilaymiz
BASE_DIR = os.path.dirname(os.path.abspath(__file__))
DB_PATH = os.path.join(BASE_DIR, "vector_db") # Ingest.py dagi bilan bir xil bo'lishi shart
DATA_PATH = os.path.join(BASE_DIR, "data")
# st.sidebar.write("### Tizim holati:")
# # 1. Avval 'data' papkasida PDF-lar borligini tekshiramiz
# if not os.path.exists(DATA_PATH) or not os.listdir(DATA_PATH):
# st.sidebar.error("❌ 'data/' papkasida PDF fayllar topilmadi!")
# else:
# st.sidebar.success(f"📂 {len(os.listdir(DATA_PATH))} ta PDF fayl topildi.")
# # 2. Bazani tekshirish va yaratish (CHROMA emas, VECTOR_DB)
# if not os.path.exists(DB_PATH):
# st.sidebar.warning("⚠️ Baza topilmadi. Ingest boshlanmoqda...")
# with st.spinner("Bilimlar bazasi yaratilmoqda..."):
# try:
# run_ingestion()
# st.success("✅ Ingestion yakunlandi!")
# st.rerun() # Bazani ko'rishi uchun refresh
# except Exception as e:
# st.error(f"❌ Ingestion xatosi: {e}")
# else:
# st.sidebar.success(f"✅ Baza tayyor: {DB_PATH}")
# st.sidebar.write("Fayllar:", os.listdir(DB_PATH))
st.title("🤖 TechFlow Support Agent")
# Initialize session state for chat history and other flags
if "chat_history" not in st.session_state:
st.session_state.chat_history = []
if "agent" not in st.session_state:
st.session_state.agent = create_agent()
if "show_ticket_form" not in st.session_state:
st.session_state.show_ticket_form = False
# Display chat messages from history on app rerun
for message in st.session_state.chat_history:
if isinstance(message, HumanMessage):
with st.chat_message("user"):
st.markdown(message.content)
elif isinstance(message, AIMessage):
with st.chat_message("assistant"):
st.markdown(message.content)
# Logic to handle ticket creation form
def submit_ticket():
summary = st.session_state.ticket_summary
desc = st.session_state.ticket_desc
email = st.session_state.ticket_email
name = st.session_state.ticket_name
if summary and desc and email and name:
with st.spinner("Creating ticket..."):
result = create_github_issue(summary, desc, email, name)
st.success(result)
st.session_state.show_ticket_form = False
# Add system message about ticket creation
st.session_state.chat_history.append(AIMessage(content=f"Ticket created: {summary}"))
else:
st.error("Please fill all fields.")
# React to user input
if prompt := st.chat_input("How can I help you today?"):
# Reset ticket form state on new query
st.session_state.show_ticket_form = False
# Display user message
st.chat_message("user").markdown(prompt)
st.session_state.chat_history.append(HumanMessage(content=prompt))
# Display assistant response
with st.chat_message("assistant"):
with st.spinner("Thinking..."):
try:
response = st.session_state.agent.invoke({
"input": prompt,
"chat_history": st.session_state.chat_history
})
output_text = response["output"]
st.markdown(output_text)
st.session_state.chat_history.append(AIMessage(content=output_text))
# Check if we should show ticket button
if "could not find the answer" in output_text.lower() or "not found" in output_text.lower():
st.session_state.show_ticket_form = True
st.rerun()
except Exception as e:
error_msg = f"An error occurred: {str(e)}"
st.error(error_msg)
st.session_state.chat_history.append(AIMessage(content=error_msg))
# Dedicated section for ticket creation if flag is set
if st.session_state.show_ticket_form:
st.divider()
st.warning("I couldn't find an answer. Would you like to raise a support ticket?")
with st.form("ticket_form"):
st.text_input("Name", key="ticket_name")
st.text_input("Email", key="ticket_email")
st.text_input("Issue Summary", key="ticket_summary")
st.text_area("Description", key="ticket_desc")
st.form_submit_button("Create Ticket", on_click=submit_ticket)
|