Update app.py

app.py

@@ -1,5 +1,5 @@
 """
-🤖 نظام RAG المبسط لـ HuggingFace Spaces - بدون pickle5
+🤖 نظام RAG لـ HuggingFace - نسخة معدلة لتعمل مع طريقة HuggingFace
 """
 
 import os
@@ -11,16 +11,31 @@ from pypdf import PdfReader
 from sentence_transformers import SentenceTransformer
 import streamlit as st
 
-# تحميل موارد NLTK مرة واحدة
-@st.cache_resource
-def load_nltk():
-    try:
-        nltk.download('punkt', quiet=True)
-        nltk.download('punkt_tab', quiet=True)
-    except:
-        pass
-    return True
+# ==================== تهيئة النظام في حالة global ====================
+# بدلاً من الاعتماد على session_state فقط
 
+_rag_system = None
+_current_file = None
+
+def get_rag_system():
+    """الحصول على أو إنشاء مثيل النظام"""
+    global _rag_system
+    if _rag_system is None:
+        _rag_system = FlowRAGSystem()
+        _rag_system.initialize()
+    return _rag_system
+
+def get_current_file():
+    """الحصول على الملف الحالي"""
+    global _current_file
+    return _current_file
+
+def set_current_file(filename):
+    """تعيين الملف الحالي"""
+    global _current_file
+    _current_file = filename
+
+# ==================== فئة النظام ====================
 class FlowRAGSystem:
     def __init__(self):
         self.model = None
@@ -31,6 +46,13 @@ class FlowRAGSystem:
     def initialize(self):
         """تهيئة النظام"""
         try:
+            # تحميل موارد NLTK
+            try:
+                nltk.download('punkt', quiet=True)
+                nltk.download('punkt_tab', quiet=True)
+            except:
+                pass
+            
             self.model = SentenceTransformer(
                 "sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2"
             )
@@ -53,9 +75,6 @@ class FlowRAGSystem:
                 reader = PdfReader(pdf_path)
                 pages_data = []
                 
-                progress_bar = st.progress(0)
-                total_pages = len(reader.pages)
-                
                 for i, page in enumerate(reader.pages):
                     text = page.extract_text()
                     if text and text.strip():
@@ -63,12 +82,6 @@ class FlowRAGSystem:
                             'page': i + 1,
                             'text': text.strip()
                         })
-                    
-                    # تحديث شريط التقدم
-                    if total_pages > 0:
-                        progress_bar.progress((i + 1) / total_pages)
-                
-                progress_bar.empty()
             
             if not pages_data:
                 st.error("❌ لم يتم العثور على نص في الملف")
@@ -81,9 +94,9 @@ class FlowRAGSystem:
                 for page in pages_data:
                     words = page['text'].split()
                     
-                    # تقسيم إلى أجزاء 250 كلمة مع تداخل 50
-                    chunk_size = 250
-                    overlap = 50
+                    # تقسيم إلى أجزاء 200 كلمة مع تداخل 40
+                    chunk_size = 200
+                    overlap = 40
                     
                     start = 0
                     while start < len(words):
@@ -157,80 +170,9 @@ class FlowRAGSystem:
             st.error(f"❌ خطأ في البحث: {e}")
             return []
 
-# واجهة Streamlit
-def main():
-    # تحميل NLTK
-    load_nltk()
-    
-    # إعداد الصفحة
-    st.set_page_config(
-        page_title="نظام RAG للمستندات",
-        page_icon="🤖",
-        layout="wide",
-        initial_sidebar_state="expanded"
-    )
-    
-    # CSS مخصص
-    st.markdown("""
-    <style>
-    .main-header {
-        text-align: center;
-        padding: 2rem;
-        background: linear-gradient(135deg, #667eea 0%, #764ba2 100%);
-        color: white;
-        border-radius: 15px;
-        margin-bottom: 2rem;
-    }
-    .result-card {
-        background: #f8f9fa;
-        border-radius: 10px;
-        padding: 1.5rem;
-        margin: 1rem 0;
-        border-left: 5px solid #4CAF50;
-        box-shadow: 0 2px 4px rgba(0,0,0,0.1);
-    }
-    .similarity-high { color: #28a745; font-weight: bold; }
-    .similarity-medium { color: #ffc107; font-weight: bold; }
-    .similarity-low { color: #dc3545; font-weight: bold; }
-    
-    .stButton>button {
-        background: linear-gradient(135deg, #667eea 0%, #764ba2 100%);
-        color: white;
-        border: none;
-        padding: 0.75rem 1.5rem;
-        border-radius: 8px;
-        font-weight: bold;
-        transition: all 0.3s;
-    }
-    .stButton>button:hover {
-        transform: translateY(-2px);
-        box-shadow: 0 4px 8px rgba(0,0,0,0.2);
-    }
-    </style>
-    """, unsafe_allow_html=True)
-    
-    # العنوان
-    st.markdown("""
-    <div class="main-header">
-        <h1>🤖 نظام RAG الذكي للمستندات</h1>
-        <p>بحث دلالي متقدم في ملفات PDF - يدعم العربية والإنجليزية</p>
-    </div>
-    """, unsafe_allow_html=True)
-    
-    # تهيئة النظام في session state
-    if 'rag_system' not in st.session_state:
-        st.session_state.rag_system = FlowRAGSystem()
-        if st.session_state.rag_system.initialize():
-            st.sidebar.success("✅ النظام جاهز للاستخدام")
-        else:
-            st.sidebar.error("❌ خطأ في تهيئة النظام")
-    
-    if 'current_file' not in st.session_state:
-        st.session_state.current_file = None
-    
-    rag_system = st.session_state.rag_system
-    
-    # الشريط الجانبي
+# ==================== الوظائف المساعدة ====================
+def create_sidebar():
+    """إنشاء الشريط الجانبي"""
     with st.sidebar:
         st.header("📁 رفع المستند")
         
@@ -242,10 +184,10 @@ def main():
         
         if uploaded_file is not None:
             if st.button("🚀 معالجة المستند", type="primary", use_container_width=True):
-                with st.spinner("جاري معالجة المستند..."):
-                    if rag_system.process_pdf(uploaded_file.getvalue(), uploaded_file.name):
-                        st.session_state.current_file = uploaded_file.name
-                        st.rerun()
+                rag_system = get_rag_system()
+                if rag_system.process_pdf(uploaded_file.getvalue(), uploaded_file.name):
+                    set_current_file(uploaded_file.name)
+                    st.rerun()
         
         st.divider()
         
@@ -267,88 +209,92 @@ def main():
         
         for question in example_questions:
             if st.button(question, use_container_width=True):
-                st.session_state.last_question = question
+                # تخزين السؤال في query params بدلاً من session state
+                st.query_params["question"] = question
                 st.rerun()
         
         st.divider()
         
         # معلومات النظام
+        rag_system = get_rag_system()
         if rag_system.chunks:
             st.header("📊 معلومات النظام")
             st.metric("الأجزاء النصية", len(rag_system.chunks))
             if rag_system.index:
                 st.metric("المتجهات", rag_system.index.ntotal)
     
-    # المنطقة الرئيسية
+    return top_k
+
+def create_main_content(top_k):
+    """إنشاء المحتوى الرئيسي"""
+    rag_system = get_rag_system()
+    current_file = get_current_file()
+    
+    # العنوان
+    st.markdown("""
+    <div style="text-align: center; padding: 2rem; background: linear-gradient(135deg, #667eea 0%, #764ba2 100%); 
+    color: white; border-radius: 15px; margin-bottom: 2rem;">
+        <h1>🤖 نظام RAG الذكي للمستندات</h1>
+        <p>بحث دلالي متقدم في ملفات PDF - يدعم العربية والإنجليزية</p>
+    </div>
+    """, unsafe_allow_html=True)
+    
     col1, col2 = st.columns([3, 1])
     
     with col1:
         st.header("💬 اسأل عن المستند")
         
         # عرض اسم الملف الحالي
-        if st.session_state.current_file:
-            st.info(f"📄 الملف الحالي: **{st.session_state.current_file}**")
+        if current_file:
+            st.info(f"📄 الملف الحالي: **{current_file}**")
         
-        # حقل السؤال
-        default_question = st.session_state.get('last_question', '')
+        # الحصول على السؤال من query params أو حقل الإدخال
+        default_question = st.query_params.get("question", "")
         question = st.text_area(
             "اكتب سؤالك هنا",
             value=default_question,
             height=120,
-            placeholder="مثال: ما هي حالة التدفق؟ أو What is flow state?\nيمكنك استخدام العربية أو الإنجليزية..."
+            placeholder="مثال: ما هي حالة التدفق؟ أو What is flow state?\nيمكنك استخدام العربية أو الإنجليزية...",
+            key="question_input"
         )
         
         # زر البحث
-        col_btn1, col_btn2 = st.columns(2)
-        with col_btn1:
-            if st.button("🔍 ابحث في المستند", type="primary", use_container_width=True):
-                if not question:
-                    st.warning("⚠️ يرجى إدخال سؤال")
-                elif rag_system.index is None:
-                    st.error("❌ يرجى معالجة مستند أولاً (من الشريط الجانبي)")
-                else:
-                    with st.spinner("جاري البحث في المستند..."):
-                        results = rag_system.search(question, top_k=top_k)
+        if st.button("🔍 ابحث في المستند", type="primary", use_container_width=True):
+            if not question:
+                st.warning("⚠️ يرجى إدخال سؤال")
+            elif rag_system.index is None:
+                st.error("❌ يرجى معالجة مستند أولاً (من الشريط الجانبي)")
+            else:
+                with st.spinner("جاري البحث في المستند..."):
+                    results = rag_system.search(question, top_k=top_k)
+                    
+                    if results:
+                        st.success(f"✅ تم العثور على {len(results)} نتيجة")
                         
-                        if results:
-                            st.success(f"✅ تم العثور على {len(results)} نتيجة")
+                        for i, result in enumerate(results):
+                            # تحديد لون التشابه
+                            similarity_score = float(result['similarity'].replace('%', '')) / 100
+                            sim_class = "similarity-high" if similarity_score >= 0.5 else \
+                                       "similarity-medium" if similarity_score >= 0.3 else \
+                                       "similarity-low"
                             
-                            for i, result in enumerate(results):
-                                # تحديد لون التشابه
-                                similarity_score = float(result['similarity'].replace('%', '')) / 100
-                                if similarity_score >= 0.5:
-                                    sim_class = "similarity-high"
-                                elif similarity_score >= 0.3:
-                                    sim_class = "similarity-medium"
-                                else:
-                                    sim_class = "similarity-low"
-                                
-                                # عرض البطاقة
-                                with st.container():
-                                    st.markdown(f"""
-                                    <div class="result-card">
-                                        <h4>🏆 النتيجة #{i+1}</h4>
-                                        <p>
-                                            <span class="{sim_class}">التشابه: {result['similarity']}</span> | 
-                                            📖 الصفحة: {result['page']} | 
-                                            🔢 الكلمات: {result['words']}
-                                        </p>
-                                        <hr>
-                                        <p>{result['text'][:400]}...</p>
-                                    </div>
-                                    """, unsafe_allow_html=True)
-                                    
-                                    # زر لعرض النص الكامل
-                                    with st.expander("📖 عرض النص الكامل"):
-                                        st.write(result['text'])
-                        else:
-                            st.error("❌ لم أجد نتائج ذات صلة في المستند")
-        
-        with col_btn2:
-            if st.button("🧹 مسح النتائج", use_container_width=True):
-                if 'last_question' in st.session_state:
-                    del st.session_state['last_question']
-                st.rerun()
+                            # عرض البطاقة
+                            st.markdown(f"""
+                            <div style="background: #f8f9fa; border-radius: 10px; padding: 1.5rem; 
+                            margin: 1rem 0; border-left: 5px solid #4CAF50; box-shadow: 0 2px 4px rgba(0,0,0,0.1);">
+                                <h4>🏆 النتيجة #{i+1}</h4>
+                                <p>
+                                    <span style="color: {'#28a745' if similarity_score >= 0.5 else '#ffc107' if similarity_score >= 0.3 else '#dc3545'}; 
+                                    font-weight: bold;">التشابه: {result['similarity']}</span> | 
+                                    📖 الصفحة: {result['page']} | 
+                                    🔢 الكلمات: {result['words']}
+                                </p>
+                                <hr>
+                                <p>{result['text'][:400]}...</p>
+                            </div>
+                            """, unsafe_allow_html=True)
+                    else:
+                        st.error("❌ لم أجد نتائج ذات صلة في المستند")
     
     with col2:
         st.header("🎯 نصائح البحث")
@@ -365,9 +311,6 @@ def main():
         
         🔸 **اطلب التفاصيل**  
         "ما هي العناصر الثمانية للتدفق؟"
-        
-        🔸 **استخدم أمثلة**  
-        "اذكر أمثلة على أنشطة تدفق"
         """)
         
         st.divider()
@@ -377,26 +320,61 @@ def main():
         **التقنيات الم��تخدمة:**
         
         • 🤖 **Sentence Transformers**  
-          نماذج embedding متعددة اللغات
-        
         • ⚡ **FAISS**  
-          بحث سريع في المتجهات
-        
         • 📄 **PyPDF**  
-          معالجة ملفات PDF
-        
-        • 🌐 **Streamlit**  
-          واجهة مستخدم تفاعلية
+        • 🌐 **Streamlit**
         """)
+
+# ==================== التطبيق الرئيسي ====================
+def main():
+    """الدالة الرئيسية للتطبيق"""
+    # إعداد الصفحة
+    st.set_page_config(
+        page_title="نظام RAG للمستندات",
+        page_icon="🤖",
+        layout="wide",
+        initial_sidebar_state="expanded"
+    )
+    
+    # CSS مخصص
+    st.markdown("""
+    <style>
+    .stButton>button {
+        background: linear-gradient(135deg, #667eea 0%, #764ba2 100%);
+        color: white;
+        border: none;
+        padding: 0.75rem 1.5rem;
+        border-radius: 8px;
+        font-weight: bold;
+    }
+    .stButton>button:hover {
+        box-shadow: 0 4px 8px rgba(0,0,0,0.2);
+    }
+    </style>
+    """, unsafe_allow_html=True)
+    
+    # إنشاء الواجهة
+    top_k = create_sidebar()
+    create_main_content(top_k)
     
     # تذييل الصفحة
     st.divider()
     st.markdown("""
-    <div style="text-align: center; color: #666; padding: 2rem;">
+    <div style="text-align: center; color: #666; padding: 1rem;">
         <p>🤖 نظام RAG للمستندات | إصدار HuggingFace Spaces</p>
-        <p>تقنية: FAISS + Sentence Transformers + Streamlit | يدعم العربية والإنجليزية</p>
+        <p>تقنية: FAISS + Sentence Transformers | يدعم العربية والإنجليزية</p>
     </div>
     """, unsafe_allow_html=True)
 
+# ==================== نقطة الدخول للتشغيل ====================
 if __name__ == "__main__":
-    main()
+    # هذا يحل مشكلة HuggingFace مع طريقة التشغيل
+    import sys
+    
+    # تحقق إذا كان يعمل في HuggingFace
+    if "huggingface" in sys.argv[0] or "spaces" in sys.argv[0]:
+        # تشغيل التطبيق مباشرة
+        main()
+    else:
+        # للتشغيل المحلي
+        main()