Update app.py

app.py

@@ -17,7 +17,7 @@ api_key = os.getenv("GOOGLE_API_KEY")
 genai.configure(api_key=api_key)
 
 
-# ✅ Function to read all PDF files (Farsi + English)
+# ✅ Read all PDFs (supports Farsi + English)
 def get_pdf_text(pdf_docs):
     text = ""
     for pdf in pdf_docs:
@@ -29,87 +29,65 @@ def get_pdf_text(pdf_docs):
     return text
 
 
-# ✅ Function to split text into chunks
+# ✅ Split text into chunks
 def get_text_chunks(text):
     splitter = RecursiveCharacterTextSplitter(chunk_size=1000, chunk_overlap=200)
-    chunks = splitter.split_text(text)
-    return chunks
+    return splitter.split_text(text)
 
 
-# ✅ Function to get embeddings for each chunk and save to vector store
+# ✅ Get embeddings + save vector store
 def get_vector_store(chunks):
-    try:
-        embeddings = GoogleGenerativeAIEmbeddings(model="models/text-embedding-004")
-        vector_store = FAISS.from_texts(chunks, embedding=embeddings)
-        vector_store.save_local("faiss_index")
-    except Exception as e:
-        raise RuntimeError(f"Error creating vector store: {e}")
+    embeddings = GoogleGenerativeAIEmbeddings(model="models/text-embedding-004")
+    vector_store = FAISS.from_texts(chunks, embedding=embeddings)
+    vector_store.save_local("faiss_index")
 
 
-# ✅ Function to get conversational chain
+# ✅ Conversational chain
 def get_conversational_chain():
     prompt_template = """
-    You are a helpful assistant. Do NOT reveal your identity (Gemini) or the company (Google).
-    Answer the question as detailed as possible using ONLY the provided context. 
-    If the answer is not in the context, say: "answer is not available in the context".
-    Do not make up answers.
+    شما یک دستیار هوشمند هستید. لطفاً فقط بر اساس متن موجود در زمینه پاسخ دهید. 
+    اگر پاسخ دقیق در متن نبود، بگویید: "اطلاعات کافی در متن موجود نیست".
+    از اضافه‌گویی یا ساختن جواب خودداری کنید.
 
-    Context:
+    --- زمینه:
     {context}
 
-    Question:
+    --- سوال:
     {question}
 
-    Answer:
+    --- پاسخ:
     """
-    try:
-        model = ChatGoogleGenerativeAI(model="gemini-2.5-pro", client=genai, temperature=0.3)
-        prompt = PromptTemplate(template=prompt_template, input_variables=["context", "question"])
-        chain = load_qa_chain(llm=model, chain_type="stuff", prompt=prompt)
-        return chain
-    except Exception as e:
-        raise RuntimeError(f"Error creating conversational chain: {e}")
+    model = ChatGoogleGenerativeAI(model="gemini-2.5-pro", client=genai, temperature=0.3)
+    prompt = PromptTemplate(template=prompt_template, input_variables=["context", "question"])
+    chain = load_qa_chain(llm=model, chain_type="map_reduce", prompt=prompt)
+    return chain
 
 
-# ✅ Function to clear chat history
+# ✅ Clear history
 def clear_chat_history():
     st.session_state.messages = [{"role": "assistant", "content": "در خدمتیم"}]
 
 
-# ✅ Function to handle user input
+# ✅ Handle user input
 def user_input(user_question):
-    try:
-        embeddings = GoogleGenerativeAIEmbeddings(model="models/text-embedding-004")
-        new_db = FAISS.load_local("faiss_index", embeddings, allow_dangerous_deserialization=True)
-        docs = new_db.similarity_search(user_question, k=4)
-        chain = get_conversational_chain()
-        response = chain({"input_documents": docs, "question": user_question}, return_only_outputs=True)
-        return response
-    except Exception as e:
-        raise RuntimeError(f"Error while answering: {e}")
-
-
-# ✅ Main function to run the Streamlit app
+    embeddings = GoogleGenerativeAIEmbeddings(model="models/text-embedding-004")
+    new_db = FAISS.load_local("faiss_index", embeddings, allow_dangerous_deserialization=True)
+    docs = new_db.similarity_search(user_question, k=8)  # ⬅️ بیشتر شده
+    chain = get_conversational_chain()
+    response = chain({"input_documents": docs, "question": user_question}, return_only_outputs=True)
+    return response
+
+
+# ✅ Main Streamlit app
 def main():
-    st.set_page_config(
-        page_title="Chatbot",
-        layout="wide",
-        initial_sidebar_state="expanded"
-    )
+    st.set_page_config(page_title="Chatbot", layout="wide", initial_sidebar_state="expanded")
 
-    # Dark theme styling
     st.markdown(
         """
         <style>
-        body {
-            background-color: #000000;
-            color: #ffffff;
-        }
-        .main {
-            background-color: #333333;
-            padding: 20px;
-            border-radius: 10px;
-        }
+        body { background-color: #000000; color: #ffffff; }
+        .main { background-color: #333333; padding: 20px; border-radius: 10px; }
+        .rtl { direction: rtl; text-align: right; }
         </style>
         """,
         unsafe_allow_html=True
@@ -120,25 +98,22 @@ def main():
 
     if not st.session_state.uploaded:
         # Upload Page
-        st.title("Your personal assistant ...")
-        pdf_docs = st.file_uploader("فایل پی دی اف مورد نظر را آپلود کنید ", accept_multiple_files=True)
+        st.title("دستیار شخصی شما ...")
+        pdf_docs = st.file_uploader("فایل(های) PDF را آپلود کنید", accept_multiple_files=True)
         if st.button("تایید"):
             if pdf_docs:
-                try:
-                    st.info("در حال پردازش ...")
-                    raw_text = get_pdf_text(pdf_docs)
-                    text_chunks = get_text_chunks(raw_text)
-                    get_vector_store(text_chunks)
-                    st.session_state.uploaded = True
-                    st.success("پردازش موفق شد ✅")
-                except RuntimeError as e:
-                    st.error(str(e))
+                st.info("در حال پردازش ...")
+                raw_text = get_pdf_text(pdf_docs)
+                text_chunks = get_text_chunks(raw_text)
+                get_vector_store(text_chunks)
+                st.session_state.uploaded = True
+                st.success("پردازش موفق شد ✅")
             else:
-                st.error("لطفا حداقل یک فایل را آپلود کنید")
+                st.error("لطفاً حداقل یک فایل انتخاب کنید")
     else:
         # Chat Page
-        st.title("Assistant ready ...")
-        st.write("میتونین سوالتونو بپرسین 👇")
+        st.title("دستیار آماده است ...")
+        st.write("سؤالتان را بپرسید 👇")
 
         if st.button("بازگشت به صفحه آپلود"):
             st.session_state.uploaded = False
@@ -153,44 +128,28 @@ def main():
         # Show history
         for message in st.session_state.messages:
             with st.chat_message(message["role"]):
-                st.markdown(
-                    f"""
-                    <div style="direction: rtl; text-align: right; font-size: 16px;">
-                        {message["content"]}
-                    </div>
-                    """,
-                    unsafe_allow_html=True
-                )
+                # تشخیص ساده فارسی برای نمایش راست به چپ
+                if any("\u0600" <= ch <= "\u06FF" for ch in message["content"]):
+                    st.markdown(f"<div class='rtl'>{message['content']}</div>", unsafe_allow_html=True)
+                else:
+                    st.write(message["content"])
 
         if prompt := st.chat_input():
             st.session_state.messages.append({"role": "user", "content": prompt})
             with st.chat_message("user"):
-                st.markdown(
-                    f"""
-                    <div style="direction: rtl; text-align: right; font-size: 16px;">
-                        {prompt}
-                    </div>
-                    """,
-                    unsafe_allow_html=True
-                )
+                st.markdown(f"<div class='rtl'>{prompt}</div>", unsafe_allow_html=True) if any(
+                    "\u0600" <= ch <= "\u06FF" for ch in prompt) else st.write(prompt)
 
             if st.session_state.messages[-1]["role"] != "assistant":
-                try:
-                    with st.chat_message("assistant"):
-                        response = user_input(prompt)
-                        if response:
-                            full_response = response['output_text']
-                            st.markdown(
-                                f"""
-                                <div style="direction: rtl; text-align: right; font-size: 16px;">
-                                    {full_response}
-                                </div>
-                                """,
-                                unsafe_allow_html=True
-                            )
-                            st.session_state.messages.append({"role": "assistant", "content": full_response})
-                except RuntimeError as e:
-                    st.error(str(e))
+                with st.chat_message("assistant"):
+                    response = user_input(prompt)
+                    if response:
+                        full_response = response['output_text']
+                        if any("\u0600" <= ch <= "\u06FF" for ch in full_response):
+                            st.markdown(f"<div class='rtl'>{full_response}</div>", unsafe_allow_html=True)
+                        else:
+                            st.write(full_response)
+                        st.session_state.messages.append({"role": "assistant", "content": full_response})
 
 
 if __name__ == "__main__":