Update app.py

app.py

@@ -10,12 +10,11 @@ import time  # Import time module for adding delays
 logging.basicConfig(level=logging.INFO, format="%(asctime)s - %(levelname)s - %(message)s")
 logger = logging.getLogger(__name__)
 
-# Initialize the Gradio client
-client = Client("MiniMaxAI/MiniMax-Text-01")
-
 # Function to call the API and get the result
 def call_api(prompt):
     try:
+        # Reload the Gradio client for each chunk
+        client = Client("MiniMaxAI/MiniMax-Text-01")
         logger.info(f"Calling API with prompt: {prompt[:100]}...")  # Log the first 100 chars of the prompt
         result = client.predict(
             message=prompt,
@@ -30,7 +29,7 @@ def call_api(prompt):
         logger.error(f"API call failed: {e}")
         raise gr.Error(f"API call failed: {str(e)}")
 
-# Function to segment the text file into chunks of 3000 words
+# Function to segment the text file into chunks of 1500 words
 def segment_text(file_path):
     try:
         logger.info(f"Reading file: {file_path}")
@@ -48,9 +47,9 @@ def segment_text(file_path):
         logger.error(f"Failed to read file: {e}")
         raise gr.Error(f"Failed to read file: {str(e)}")
     
-    # Split the text into chunks of 3000 words
+    # Split the text into chunks of 1500 words
     words = text.split()
-    chunks = [" ".join(words[i:i + 3000]) for i in range(0, len(words), 3000)]
+    chunks = [" ".join(words[i:i + 1500]) for i in range(0, len(words), 1500)]
     logger.info(f"Segmented text into {len(chunks)} chunks.")
     return chunks