Update app.py

app.py

@@ -1,64 +1,139 @@
-import streamlit as st
+import os
 import time
-from langchain.document_loaders import PyPDFLoader
-from langchain.text_splitter import RecursiveCharacterTextSplitter
+
 from langchain_together import TogetherEmbeddings
-from langchain.vectorstores import FAISS
+import streamlit as st
 from langchain.chat_models import ChatOpenAI
+from langchain.document_loaders import PyPDFLoader
+from langchain.text_splitter import RecursiveCharacterTextSplitter
+from langchain.schema import Document
 from langchain.chains import RetrievalQA
+from langchain.vectorstores import FAISS
 from langchain.indexes import VectorstoreIndexCreator
 
-
+# ----------------- تنظیمات صفحه -----------------
+st.set_page_config(page_title="چت‌ بات توانا", page_icon="🪖", layout="wide")
+
+st.markdown("""
+    <style>
+    @import url('https://fonts.googleapis.com/css2?family=Vazirmatn:wght@400;700&display=swap');
+    html, body, [class*="css"] {
+        font-family: 'Vazirmatn', Tahoma, sans-serif;
+        direction: rtl;
+        text-align: right;
+    }
+    .stApp {
+        background: url("./military_bg.jpeg") no-repeat center center fixed;
+        background-size: cover;
+        backdrop-filter: blur(2px);
+    }
+    .stChatMessage {
+        background-color: rgba(255,255,255,0.8);
+        border: 1px solid #4e8a3e;
+        border-radius: 12px;
+        padding: 16px;
+        margin-bottom: 15px;
+        box-shadow: 0 4px 10px rgba(0,0,0,0.2);
+        animation: fadeIn 0.4s ease-in-out;
+    }
+    .stTextInput > div > input, .stTextArea textarea {
+        background-color: rgba(255,255,255,0.9) !important;
+        border-radius: 8px !important;
+        direction: rtl;
+        text-align: right;
+        font-family: 'Vazirmatn', Tahoma;
+    }
+    .stButton>button {
+        background-color: #4e8a3e !important;
+        color: white !important;
+        font-weight: bold;
+        border-radius: 10px;
+        padding: 8px 20px;
+        transition: 0.3s;
+    }
+    .stButton>button:hover {
+        background-color: #3c6d30 !important;
+    }
+    .header-text {
+        text-align: center;
+        margin-top: 20px;
+        margin-bottom: 40px;
+        background-color: rgba(255, 255, 255, 0.75);
+        padding: 20px;
+        border-radius: 20px;
+        box-shadow: 0 4px 12px rgba(0,0,0,0.2);
+    }
+    .header-text h1 {
+        font-size: 42px;
+        color: #2c3e50;
+        margin: 0;
+        font-weight: bold;
+    }
+    .subtitle {
+        font-size: 18px;
+        color: #34495e;
+        margin-top: 8px;
+    }
+    @keyframes fadeIn {
+        from { opacity: 0; transform: translateY(10px); }
+        to { opacity: 1; transform: translateY(0); }
+    }
+    </style>
+""", unsafe_allow_html=True)
+
+col1, col2, col3 = st.columns([1, 0.2, 1])
+with col2:
+    st.image("army.png", width=240)
+
+st.markdown("""
+    <div class="header-text">
+        <h1>چت‌ بات توانا</h1>
+        <div class="subtitle">دستیار هوشمند</div>
+    </div>
+""", unsafe_allow_html=True)
+
+# ----------------- لود PDF و ساخت ایندکس -----------------
 @st.cache_resource
-def load_chunks_and_embeddings():
-    with st.spinner("در حال بارگذاری فایل و آماده‌سازی... لطفاً صبور باشید 🙏"):
-        progress_bar = st.progress(0, text="در حال بارگذاری فایل PDF...")
-
+def get_pdf_index():
+    with st.spinner('📄 در حال پردازش فایل PDF...'):
         pdf_loader = PyPDFLoader('test1.pdf')
-        pages = pdf_loader.load()
-        progress_bar.progress(30, text="صفحات PDF بارگذاری شد. در حال ایجاد مدل برداری...")
 
         embeddings = TogetherEmbeddings(
             api_key="0291f33aee03412a47fa5d8e562e515182dcc5d9aac5a7fb5eefdd1759005979"
         )
-        progress_bar.progress(60, text="مدل Embedding ساخته شد. در حال ایجاد ایندکس...")
 
-        index = VectorstoreIndexCreator(
-            embedding=embeddings,
-            text_splitter=RecursiveCharacterTextSplitter(chunk_size=300, chunk_overlap=0)
-        ).from_loaders([pdf_loader])
+        index = VectorstoreIndexCreator(embedding=embeddings, text_splitter=RecursiveCharacterTextSplitter(chunk_size=300, chunk_overlap=0)).from_loaders([pdf_loader])
+        
+        st.write(f"تعداد بخش‌های پردازش‌شده: {len(index.vectorstore)}")
+        return index
+
+# ----------------- بارگذاری دیتا -----------------
+index = get_pdf_index()
 
-        progress_bar.progress(100, text="بارگذاری کامل شد! ✅")
-    return index
 llm = ChatOpenAI(
     base_url="https://api.together.xyz/v1",
     api_key='0291f33aee03412a47fa5d8e562e515182dcc5d9aac5a7fb5eefdd1759005979',
-    model="meta-llama/Llama-3-70B-Instruct-Turbo-Free"
+    model="meta-llama/Llama-3.3-70B-Instruct-Turbo-Free"
 )
 
-index = load_chunks_and_embeddings()
-
 chain = RetrievalQA.from_chain_type(
-  llm=llm,
-  chain_type='stuff',
-  retriever=index.vectorstore.as_retriever(), input_key='question')
-
+    llm=llm,
+    chain_type='stuff',
+    retriever=index.vectorstore.as_retriever(),
+    input_key='question'
+)
 
-# --- 💬 چت بات
 if 'messages' not in st.session_state:
     st.session_state.messages = []
 
 if 'pending_prompt' not in st.session_state:
     st.session_state.pending_prompt = None
 
-st.title("📄🤖 دستیار PDF شما")
-
-# نمایش تاریخچه گفتگو
 for msg in st.session_state.messages:
     with st.chat_message(msg['role']):
         st.markdown(f"🗨️ {msg['content']}", unsafe_allow_html=True)
 
-prompt = st.chat_input("سوالی از PDF داری؟")
+prompt = st.chat_input("چطور می‌تونم کمک کنم؟")
 
 if prompt:
     st.session_state.messages.append({'role': 'user', 'content': prompt})
@@ -70,8 +145,7 @@ if st.session_state.pending_prompt:
         thinking = st.empty()
         thinking.markdown("🤖 در حال فکر کردن...")
 
-        # اجرای جستجو در ایندکس
-        response = chain.run(f'فقط به زبان فارسی جواب بده. سوال: {st.session_state.pending_prompt}')
+        response = chain.run(f'پاسخ را فقط به زبان فارسی جواب بده. سوال: {st.session_state.pending_prompt}')
         answer = response.split("Helpful Answer:")[-1].strip()
         if not answer:
             answer = "متأسفم، اطلاعات دقیقی در این مورد ندارم."