Update app.py

app.py

@@ -2,7 +2,6 @@ import os
 import time
 import streamlit as st
 from langchain.chat_models import ChatOpenAI
-
 from transformers import AutoTokenizer, AutoModel
 from langchain.document_loaders import PyPDFLoader
 from langchain.text_splitter import RecursiveCharacterTextSplitter
@@ -156,24 +155,21 @@ if prompt:
 # ----------------- پاسخ مدل -----------------
 if st.session_state.pending_prompt:
     with st.chat_message('ai'):
-        thinking = st.empty()
-        thinking.markdown("🤖 در حال فکر کردن از روی PDF...")
-
+        # اضافه کردن پروگرس بار
+        progress_bar = st.progress(0, text="در حال پردازش...")
+        
         try:
             response = chain.run(f"سوال: {st.session_state.pending_prompt}")
             answer = response.strip()
-        except Exception as e:
-            answer = f"خطا در پاسخ‌دهی: {str(e)}"
 
-        thinking.empty()
+            # شبیه سازی پردازش برای به روز کردن پروگرس بار
+            for i in range(0, 101, 20):
+                progress_bar.progress(i)
+                time.sleep(0.1)  # شبیه سازی سرعت پردازش
 
-        full_response = ""
-        placeholder = st.empty()
-        for word in answer.split():
-            full_response += word + " "
-            placeholder.markdown(full_response + "▌")
-            time.sleep(0.03)
+        except Exception as e:
+            answer = f"خطا در پاسخ‌دهی: {str(e)}"
 
-        placeholder.markdown(full_response)
-        st.session_state.messages.append({'role': 'ai', 'content': full_response})
-        st.session_state.pending_prompt = None
+        progress_bar.progress(100)  # کامل شدن پروگرس بار
+        
+        st.session_state.messages.append({'role': 'ai', 'content': answer})