Update app.py

app.py

@@ -21,8 +21,16 @@ def call_api(prompt):
 
 # Function to segment the text file into chunks of 3000 words
 def segment_text(file_path):
-    with open(file_path, "r") as f:
-        text = f.read()
+    try:
+        # Try reading with UTF-8 encoding first
+        with open(file_path, "r", encoding="utf-8") as f:
+            text = f.read()
+    except UnicodeDecodeError:
+        # Fallback to latin-1 encoding if UTF-8 fails
+        with open(file_path, "r", encoding="latin-1") as f:
+            text = f.read()
+    
+    # Split the text into chunks of 3000 words
     words = text.split()
     chunks = [" ".join(words[i:i + 3000]) for i in range(0, len(words), 3000)]
     return chunks
@@ -42,7 +50,7 @@ def process_text(file, prompt):
     # Save results as individual text files
     os.makedirs("outputs", exist_ok=True)
     for idx, result in enumerate(results):
-        with open(f"outputs/output_{idx}.txt", "w") as f:
+        with open(f"outputs/output_{idx}.txt", "w", encoding="utf-8") as f:
             f.write(result)
     
     # Upload to Hugging Face dataset
@@ -75,5 +83,5 @@ with gr.Blocks() as demo:
         outputs=[output_zip, output_message]
     )
 
-# Launch the Gradio app
-demo.launch()
+# Launch the Gradio app with a public link
+demo.launch(share=True)