Update app.py

app.py

@@ -6,12 +6,12 @@ from transformers import pipeline
 import PyPDF2
 import gradio as gr
 
-# Download and initialize required tools
+# تحميل وتفعيل الأدوات المطلوبة
 spacy.cli.download("en_core_web_sm")
 nlp = spacy.load("en_core_web_sm")
 nltk.download('punkt')
 
-# Check if GPU is available and use it
+# التحقق من توفر GPU واستخدامه
 device = 0 if torch.cuda.is_available() else -1
 analyzer = pipeline("sentiment-analysis", model="distilbert-base-uncased-finetuned-sst-2-english", device=device)
 
@@ -47,6 +47,8 @@ def extract_pdf_text(file_path):
 
 def analyze_and_complete(file_paths):
     results = []
+    output_directory = "/Volumes/CLOCKWORK T/clockworkspace/first pro"
+    
     for file_path in file_paths:
         if file_path.endswith(".pdf"):
             text = extract_pdf_text(file_path)
@@ -54,7 +56,6 @@ def analyze_and_complete(file_paths):
             with open(file_path, "r", encoding="utf-8") as file:
                 text = file.read()
         
-        output_directory = "/Users/Home/Library/Mobile Documents/com~apple~CloudDocs/osa/سيناريوهات/ليالي ألف ليلة"
         filename_prefix = os.path.splitext(os.path.basename(file_path))[0]
         
         spacy_entities = spacy_ner_analysis(text)
@@ -62,14 +63,28 @@ def analyze_and_complete(file_paths):
         quotes = nltk_extract_quotes(text)
         token_count = count_tokens(text)
         
+        # حفظ النتائج إلى ملفات
+        with open(os.path.join(output_directory, f"{filename_prefix}_spacy_entities.txt"), "w", encoding="utf-8") as file:
+            file.write(str(spacy_entities))
+
+        with open(os.path.join(output_directory, f"{filename_prefix}_sentences.txt"), "w", encoding="utf-8") as file:
+            file.write("\n".join(sentences))
+
+        with open(os.path.join(output_directory, f"{filename_prefix}_quotes.txt"), "w", encoding="utf-8") as file:
+            file.write("\n".join(quotes))
+
+        with open(os.path.join(output_directory, f"{filename_prefix}_token_count.txt"), "w", encoding="utf-8") as file:
+            file.write(str(token_count))
+        
         results.append((str(spacy_entities), "\n".join(sentences), "\n".join(quotes), str(token_count)))
+    
     return results
 
-# Define the Gradio interface
+# تعريف واجهة Gradio
 interface = gr.Interface(
     fn=analyze_and_complete,
     inputs=gr.File(file_count="multiple", type="filepath"),
-    outputs=["text", "text", "text", "text"],
+    outputs=gr.outputs.JSON(),
     title="Movie Script Analyzer and Completer",
     description="Upload text, PDF, or DOCX files to analyze and complete the movie script."
 )