|
|
import sys |
|
|
import uuid |
|
|
|
|
|
import gradio as gr |
|
|
|
|
|
from database_setup import initialize_database |
|
|
from chat_utils import create_rag_chain, format_sources, create_session_history_manager |
|
|
from langchain_core.runnables.history import RunnableWithMessageHistory |
|
|
|
|
|
print("Inicjalizacja bazy danych...") |
|
|
baza = initialize_database() |
|
|
if baza is None: |
|
|
print("Nie udało się zainicjalizować bazy danych. Zakończenie pracy.") |
|
|
sys.exit(1) |
|
|
|
|
|
rag_chain = create_rag_chain(baza) |
|
|
get_session_history = create_session_history_manager() |
|
|
conversational_rag_chain = RunnableWithMessageHistory( |
|
|
rag_chain, |
|
|
get_session_history, |
|
|
input_messages_key="input", |
|
|
history_messages_key="chat_history", |
|
|
output_messages_key="answer", |
|
|
) |
|
|
|
|
|
def respond(message, chat_history, sess_id): |
|
|
"""Obsługuje odpowiedź na wiadomość użytkownika.""" |
|
|
try: |
|
|
response = conversational_rag_chain.invoke( |
|
|
{"input": message}, |
|
|
config={"configurable": {"session_id": sess_id}}, |
|
|
) |
|
|
except Exception as e: |
|
|
chat_history.append((message, f"Błąd podczas przetwarzania: {e}")) |
|
|
return chat_history |
|
|
|
|
|
context_docs = response.get("context", []) |
|
|
|
|
|
try: |
|
|
debug_scores = baza.similarity_search_with_score(message, k=len(context_docs)) |
|
|
for i, (doc, score) in enumerate(debug_scores): |
|
|
print(f"Chunk {i+1}: similarity_score={score}, title={doc.metadata.get('title')}") |
|
|
except Exception: |
|
|
pass |
|
|
|
|
|
sources_md = format_sources(context_docs) |
|
|
answer = response.get("answer") or "" |
|
|
answer_with_sources = f"{answer}\n\n**Źródła:**\n{sources_md}" |
|
|
chat_history.append((message, answer_with_sources)) |
|
|
return chat_history |
|
|
|
|
|
with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue"), title="Szuflada Chatbot") as demo: |
|
|
session_id = gr.State(lambda: str(uuid.uuid4())) |
|
|
|
|
|
gr.Markdown( |
|
|
"# Czat z Moją Szufladą\n" |
|
|
"### Zadaj pytanie na temat treści ze strony [mojaszuflada.pl](https://mojaszuflada.pl)" |
|
|
) |
|
|
chatbot = gr.Chatbot(label="Rozmowa", height=500) |
|
|
|
|
|
with gr.Row(): |
|
|
msg = gr.Textbox( |
|
|
show_label=False, |
|
|
placeholder="Wpisz swoje pytanie...", |
|
|
container=False, |
|
|
scale=7, |
|
|
) |
|
|
submit_btn = gr.Button("Wyślij", variant="primary", scale=1) |
|
|
|
|
|
submit_btn.click(respond, [msg, chatbot, session_id], [chatbot]) \ |
|
|
.then(lambda: gr.update(value=""), None, [msg], queue=False) |
|
|
msg.submit(respond, [msg, chatbot, session_id], [chatbot]) \ |
|
|
.then(lambda: gr.update(value=""), None, [msg], queue=False) |
|
|
|
|
|
if __name__ == "__main__": |
|
|
demo.launch(inbrowser=True) |
