Sahaf.AI / app.py
HayriTalha's picture
gerekli dosyalar eklendi
112c4c0 verified
raw
history blame
1.87 kB
import streamlit as st
import nest_asyncio
from src.chatbot import create_rag_chain
nest_asyncio.apply()
# --- ARAYÜZ TASARIMI ---
st.set_page_config(page_title="Sahaf.AI",layout="centered")
# Ana başlık
st.markdown("<h1 style='text-align: center;'>SAHAF.AI</h1>", unsafe_allow_html=True)
# --- CHATBOT MANTIĞI ---
chain = create_rag_chain()
if "messages" not in st.session_state:
st.session_state.messages = []
chat_container = st.container(height=500, border=True)
with chat_container:
if not st.session_state.messages:
st.markdown("### Merhaba, Ben Sahaf.AI! 👋")
st.info(
"Kitaplar, yazarları ve eserleri hakkında merak ettiklerini "
"bana sorabilirsin. Sana yardımcı olmaktan mutluluk duyarım."
)
st.markdown("##### Örnek Sorular:")
st.markdown("- *Ahmet Ümit'in İstanbul Hatırası romanının konusu nedir?*")
st.markdown("- *Bana polisiye türünde 3 tane roman önerebilir misin?*")
else:
for message in st.session_state.messages:
with st.chat_message(message["role"]):
st.markdown(message["content"])
if prompt := st.chat_input("Bir yazar veya kitap hakkında soru sorun..."):
st.session_state.messages.append({"role": "user", "content": prompt})
# Chatbot'un cevabını oluştur
with st.spinner("Düşünüyorum..."):
try:
result = chain.invoke({"query": prompt})
response = result["result"]
except Exception as e:
response = "Üzgünüm, bir sorunla karşılaştım. Lütfen birkaç saniye sonra tekrar deneyin."
print(f"HATA OLUŞTU: {e}")
st.session_state.messages.append({"role": "assistant", "content": response})
st.rerun()