Spaces:
Sleeping
Sleeping
File size: 10,261 Bytes
92f70a2 |
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 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 |
# appv1.py
# A RAG rendszer grafikus felhasználói felülete Streamlit segítségével.
# Végleges verzió, Chat és Admin felülettel.
# Igazítva a backendv1.py-hoz.
# Kiegészítve a legjobb találati pontszám megjelenítésével.
import streamlit as st
import sys
import os
# A backendv1.py importálása, a futtatható könyvtárhoz hozzáadása.
# Feltételezi, hogy a backendv1.py és az appv1.py ugyanabban a mappában van.
sys.path.append(os.path.dirname(__file__))
# Az összes szükséges függvény importálása a backendből
from backendv1 import (
initialize_backend,
process_query,
index_feedback,
get_all_feedback,
delete_feedback_by_id,
update_feedback_comment,
CONFIG
)
# --- Oldal Konfiguráció ---
st.set_page_config(page_title="Dunaelektronika AI", layout="wide")
st.title("🤖 Dunaelektronika AI Asszisztens")
# --- Backend Betöltése (gyorsítótárazva) ---
@st.cache_resource
def load_backend_components():
return initialize_backend()
backend = load_backend_components()
# --- Session State Inicializálása ---
if "messages" not in st.session_state:
st.session_state.messages = []
if "last_confidence_score" not in st.session_state:
st.session_state.last_confidence_score = "N/A"
if "page" not in st.session_state:
st.session_state.page = "Chat"
# --- Navigáció az Oldalsávon ---
with st.sidebar:
st.header("Menü")
if st.button("💬 Chat", use_container_width=True,
type="primary" if st.session_state.page == "Chat" else "secondary"):
st.session_state.page = "Chat"
st.rerun()
if st.button("⚙️ Feedback Adminisztráció", use_container_width=True,
type="primary" if st.session_state.page == "Admin" else "secondary"):
st.session_state.page = "Admin"
st.rerun()
st.write("---")
# ==============================================================================
# = CHAT OLDAL LOGIKÁJA =
# ==============================================================================
if st.session_state.page == "Chat":
with st.sidebar:
st.header("Beállítások")
# A 0.1 egy jó alapértelmezett érték, de a pontos tartomány a Cross-Encoder modell kimenetétől függ
confidence_threshold = st.slider("Minimális pontossági küszöb", min_value=-5.0, max_value=5.0, value=0.1,
step=0.1)
fallback_message = st.text_area("Válasz alacsony pontosságnál",
"A rendelkezésre álló információk alapján sajnos nem tudok egyértelmű választ adni a kérdésre.",
height=100)
CONFIG["GENERATION_TEMPERATURE"] = st.slider("Kreativitás (Temperature)", 0.0, 1.0, 0.6, 0.05)
st.write("---")
st.subheader("Utolsó Válasz Elemzése")
score = st.session_state.last_confidence_score
if score == "N/A":
level, help_text = "N/A", "Tegyen fel egy kérdést a megbízhatóság méréséhez."
elif score is None:
level, help_text = "Alap Rangsor (RRF)", "A Cross-Encoder bizonytalan volt."
elif score == 10.0:
level, help_text = "Kurált Válasz", "Ez egy korábban megadott, pontosított válasz."
else:
help_text = f"Nyers pontszám: {score:.4f}"
if score > 1.0:
level = "Magas"
elif score >= -1.5:
level = "Közepes"
else:
level = "Alacsony"
st.metric(label="Keresési Magabiztosság", value=level, help=help_text)
# Chat Előzmények Megjelenítése
for i, message in enumerate(st.session_state.messages):
with st.chat_message(message["role"]):
st.markdown(message["content"].replace('$', '\\$'))
if message["role"] == "assistant":
# --- HOZZÁADOTT RÉSZ ---
# A válaszhoz tartozó pontszám megjelenítése, ha létezik.
score_value = message.get("score")
if score_value is not None:
if score_value == 10.0:
score_display = "Kurált válasz (legmagasabb)"
else:
score_display = f"{score_value:.4f}"
st.caption(f"A válasz legjobb score értéke: **{score_display}**")
# --- HOZZÁADOTT RÉSZ VÉGE ---
if message.get("sources"):
with st.expander("Felhasznált források"):
for source in message["sources"]:
st.caption(f"Forrás: {source.get('url', 'N/A')}")
st.markdown(f"> {source.get('content', '')[:250]}...")
feedback_key_prefix = f"feedback_{i}"
if not message.get("rated"):
st.write("---")
cols = st.columns(7)
if cols[0].button("👍 Jó", key=f"{feedback_key_prefix}_good"):
message["rated"] = "good";
st.toast("Köszönjük a visszajelzést!");
st.rerun()
if cols[1].button("👎 Rossz", key=f"{feedback_key_prefix}_bad"):
message["rated"] = "bad";
st.rerun()
if message.get("rated") == "bad":
with st.form(key=f"{feedback_key_prefix}_form"):
correction_text = st.text_area("Javítás:", key=f"{feedback_key_prefix}_text",
value=message.get("correction", ""))
if st.form_submit_button("Javítás elküldése"):
# Hívás a backendv1 függvényre
index_feedback(backend["es_client"], backend["embedding_model"],
message["original_question"], correction_text)
st.success("Javításodat rögzítettük!");
message["rated"] = "corrected";
st.rerun()
# Felhasználói Kérdés Feldolgozása
if prompt := st.chat_input("Kérdezz valamit a Dunaelektronikáról..."):
st.session_state.messages.append({"role": "user", "content": prompt})
with st.spinner("Keresek és gondolkodom..."):
# Hívás a backendv1 függvényre
response_data = process_query(prompt, st.session_state.messages, backend, confidence_threshold,
fallback_message)
st.session_state.last_confidence_score = response_data.get("confidence_score")
# --- MÓDOSÍTOTT RÉSZ ---
# A válasz üzenethez hozzáadjuk a 'score' kulcsot is, hogy később meg tudjuk jeleníteni.
st.session_state.messages.append({
"role": "assistant",
"content": response_data.get("answer", "Hiba történt."),
"sources": response_data.get("sources", []),
"original_question": prompt,
"rated": False,
"score": response_data.get("confidence_score") # Itt adjuk hozzá a pontszámot
})
# --- MÓDOSÍTOTT RÉSZ VÉGE ---
st.rerun()
# ==============================================================================
# = ADMIN OLDAL LOGIKÁJA =
# ==============================================================================
elif st.session_state.page == "Admin":
st.header("Rögzített Visszajelzések Kezelése")
if st.button("Lista frissítése"):
st.cache_data.clear()
@st.cache_data(ttl=60)
def get_cached_feedback():
# Hívás a backendv1 függvényre
return get_all_feedback(backend["es_client"], CONFIG["FEEDBACK_INDEX_NAME"])
feedback_list = get_cached_feedback()
if not feedback_list:
st.warning("Nincsenek rögzített visszajelzések.")
else:
st.info(f"Összesen {len(feedback_list)} visszajelzés található.")
for item in feedback_list:
doc_id = item["_id"]
source = item["_source"]
with st.container(border=True):
st.markdown(f"**Kérdés:** `{source.get('question_text', 'N/A')}`")
with st.form(key=f"edit_form_{doc_id}"):
new_comment = st.text_area("Javítás/Megjegyzés:", value=source.get('correction_text', ''),
key=f"text_{doc_id}", label_visibility="collapsed")
col1, col2 = st.columns([4, 1])
with col1:
if st.form_submit_button("💾 Mentés"):
# Hívás a backendv1 függvényre
if update_feedback_comment(backend["es_client"], CONFIG["FEEDBACK_INDEX_NAME"], doc_id,
new_comment):
st.success("Sikeresen frissítve!")
st.cache_data.clear()
st.rerun()
else:
st.error("Hiba történt a frissítés során.")
with col2:
if st.form_submit_button("🗑️ Törlés"):
# Hívás a backendv1 függvényre
if delete_feedback_by_id(backend["es_client"], CONFIG["FEEDBACK_INDEX_NAME"], doc_id):
st.success(f"Sikeresen törölve!")
st.cache_data.clear()
st.rerun()
else:
st.error("Hiba történt a törlés során.")
st.caption(f"Elasticsearch ID: {doc_id} | Időbélyeg: {source.get('timestamp', 'N/A')}") |