Update app.py

app.py

@@ -62,6 +62,19 @@ def analyze_and_complete(file_paths):
         quotes = nltk_extract_quotes(text)
         token_count = count_tokens(text)
         
+        # Save results to files
+        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
 
@@ -69,7 +82,7 @@ def analyze_and_complete(file_paths):
 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."
 )