Update app.py

app.py

@@ -57,6 +57,7 @@
 import os
 import gradio as gr
 from langchain.text_splitter import RecursiveCharacterTextSplitter
+import shutil
 
 # Ensure data folder exists
 os.makedirs("data", exist_ok=True)
@@ -65,26 +66,27 @@ def process_files(italy_file, france_file):
     docs = []
     text_splitter = RecursiveCharacterTextSplitter(chunk_size=300, chunk_overlap=50)
 
-    # Save uploaded files
-    file_paths = {}
     try:
+        file_paths = {}
+        
+        # Move italy file
         if italy_file is not None:
             italy_path = "data/italy.txt"
-            with open(italy_path, "wb") as f:
-                f.write(italy_file.read())
+            shutil.copy(italy_file.name, italy_path)
             file_paths["Italy"] = italy_path
 
+        # Move france file
         if france_file is not None:
             france_path = "data/france.txt"
-            with open(france_path, "wb") as f:
-                f.write(france_file.read())
+            shutil.copy(france_file.name, france_path)
             file_paths["France"] = france_path
+
     except Exception as e:
         return f"❌ Error while saving files: {e}", []
 
-    # Process text from files
+    # Process the files
     try:
-        for name, path in file_paths.items():
+        for country, path in file_paths.items():
             with open(path, "r", encoding="utf-8") as f:
                 content = f.read()
                 chunks = text_splitter.create_documents([content])
@@ -103,8 +105,8 @@ gr.Interface(
     ],
     outputs=[
         gr.Textbox(label="Status"),
-        gr.Textbox(label="First 10 Chunks", lines=20)
+        gr.Textbox(label="First 10 Chunks (Preview)", lines=20)
     ],
-    title="Text Chunker with LangChain",
-    description="Upload italy.txt and france.txt to split them into 300-character chunks."
+    title="Text Chunking App (LangChain)",
+    description="Upload two .txt files (Italy & France). They will be split into 300-character chunks."
 ).launch()