Update app.py

app.py

@@ -9,35 +9,90 @@ import re
 # Load model
 model = SentenceTransformer('paraphrase-multilingual-MiniLM-L12-v2')
 
-# Reference philosophical data
+# تابع برای استخراج نقل‌قول‌ها از فایل
+def load_quotes_from_file(filename, philosopher):
+    try:
+        with open(filename, 'r', encoding='utf-8') as file:
+            content = file.read()
+        
+        # حذف بخش‌های متا
+        content = re.sub(r'\[file name\].*?\[file content begin\]', '', content, flags=re.DOTALL)
+        content = re.sub(r'\[file content end\].*', '', content, flags=re.DOTALL)
+        
+        # تقسیم به جملات معنادار
+        sentences = re.split(r'(?<!\w\.\w.)(?<![A-Z][a-z]\.)(?<=\.|\?|\!)\s', content)
+        
+        # فیلتر جملات کوتاه و بی‌معنی
+        meaningful_quotes = [
+            s.strip() for s in sentences 
+            if len(s.split()) > 4  # جملات حداقل 4 کلمه‌ای
+            and not s.startswith((' ', '\n')) 
+            and len(s) < 500  # جملات خیلی طولانی نباشند
+        ]
+        
+        return meaningful_quotes
+    except Exception as e:
+        print(f"Error loading {filename}: {str(e)}")
+        return []
+
+# ساختار داده‌های فلسفی جدید
 school_data = {
-    "Stoicism": [
-        "The key to happiness is accepting things we cannot control.",
-        "Virtue is the only good.",
-        "Live according to nature."
-    ],
-    "Existentialism": [
-        "Existence precedes essence.",
-        "Man is nothing else but what he makes of himself.",
-        "Life has no inherent meaning; we create our own meaning."
-    ],
-    "Rationalism": [
-        "Reason is the chief source of knowledge.",
-        "I think, therefore I am.",
-        "Innate ideas exist prior to experience."
-    ],
-    "Empiricism": [
-        "Knowledge comes only from sensory experience.",
-        "The mind is a blank slate at birth.",
-        "All ideas originate in experience."
-    ]
+    "Hegelianism": {
+        "philosophers": ["Hegel"],
+        "quotes": []
+    },
+    "Aristotelianism": {
+        "philosophers": ["Aristotle"],
+        "quotes": []
+    },
+    "Schopenhauerian": {
+        "philosophers": ["Schopenhauer"],
+        "quotes": []
+    },
+    "Nietzschean": {
+        "philosophers": ["Nietzsche"],
+        "quotes": []
+    }
 }
 
+# تابع بارگیری نقل‌قول‌ها
+def load_quotes():
+    try:
+        school_data["Hegelianism"]["quotes"] = load_quotes_from_file("Hegel.txt", "Hegel")
+        school_data["Aristotelianism"]["quotes"] = load_quotes_from_file("Aristotle.txt", "Aristotle")
+        school_data["Schopenhauerian"]["quotes"] = load_quotes_from_file("Arthur-Schopenhauer-Quotes.txt", "Schopenhauer")
+        school_data["Nietzschean"]["quotes"] = load_quotes_from_file("Friedrich-Nietzsche.txt", "Nietzsche")
+        
+        # حذف مکاتب خالی
+        for school in list(school_data.keys()):
+            if not school_data[school]["quotes"]:
+                del school_data[school]
+        
+        print(f"Loaded quotes from {len(school_data)} philosophical schools")
+    except Exception as e:
+        print(f"Error loading quotes: {str(e)}")
+
+# فراخوانی تابع بارگیری
+load_quotes()
+
+# پروفایل مکاتب
 school_profiles = {
-    "Stoicism": {"timeline": "3rd century BCE – 3rd century CE", "profile": "Calm, Resilient, Virtue-centered"},
-    "Existentialism": {"timeline": "19th – 20th century", "profile": "Individualistic, Authentic, Anxious"},
-    "Rationalism": {"timeline": "17th – 18th century", "profile": "Logical, Abstract, Confident"},
-    "Empiricism": {"timeline": "17th – 18th century", "profile": "Practical, Observational, Experimental"}
+    "Hegelianism": {
+        "timeline": "19th century",
+        "profile": "Dialectical, Historical, Idealist"
+    },
+    "Aristotelianism": {
+        "timeline": "4th century BCE",
+        "profile": "Logical, Empirical, Teleological"
+    },
+    "Schopenhauerian": {
+        "timeline": "19th century",
+        "profile": "Pessimistic, Compassionate, Will-centered"
+    },
+    "Nietzschean": {
+        "timeline": "19th century",
+        "profile": "Existential, Will-to-Power, Übermensch"
+    }
 }
 
 psychological_categories = [
@@ -92,7 +147,7 @@ def psychological_analysis(text):
     return output
 
 def create_semantic_plot(user_vec, best_school):
-    ref_quotes = school_data[best_school]
+    ref_quotes = school_data[best_school]["quotes"]  # تغییر اینجا برای تطابق با ساختار جدید
     quote_vecs = model.encode(ref_quotes)
     labels = [f"Ref {i+1}" for i in range(len(ref_quotes))]
 
@@ -131,8 +186,8 @@ def analyze_text(text):
     user_vec = model.encode([text])[0]
     best_school, best_score, best_match = None, -1, ""
 
-    for school, quotes in school_data.items():
-        for quote in quotes:
+    for school, data in school_data.items():  # تغییر اینجا برای تطابق با ساختار جدید
+        for quote in data["quotes"]:
             quote_vec = model.encode([quote])[0]
             score = cosine_similarity([user_vec], [quote_vec])[0][0]
             if score > best_score:
@@ -172,4 +227,4 @@ with gr.Blocks(title="Philosophical Analyzer") as demo:
                      outputs=[school, score, profile_box, timeline, best_quote, conceptual_map, psych_box])
     clear_btn.click(clear_fields, outputs=[input_text, school, score, profile_box, timeline, best_quote, conceptual_map, psych_box])
 
-demo.launch()
+demo.launch()