Update app.py

app.py

@@ -1,76 +1,94 @@
 from fastapi import FastAPI, Query
+import uvicorn
 import requests
 import os
-import uvicorn
-from uuid import uuid4
 from pydub import AudioSegment
+from uuid import uuid4
 
 app = FastAPI()
 
-TTS_API_URL = "https://hivecorp-s8test.hf.space/generate"  # Replace with your TTS API URL
-SAVE_DIR = "generated_audios"
+# ✅ Create directory to store audio files
+AUDIO_DIR = "audio_files"
+os.makedirs(AUDIO_DIR, exist_ok=True)
 
-# Ensure save directory exists
-os.makedirs(SAVE_DIR, exist_ok=True)
+# ✅ Replace with your actual TTS API URL
+TTS_API_URL = "https://your-tts-api.com/generate"  # Change this
 
+
+# 🔹 Function: Splitting long text into smaller chunks
 def split_text(text, max_length=500):
-    """Splits text into smaller chunks without breaking words."""
+    """Splits text into chunks of max_length characters"""
     words = text.split()
-    chunks, chunk = [], []
+    chunks = []
+    current_chunk = []
+    current_length = 0
     
     for word in words:
-        if len(" ".join(chunk) + " " + word) <= max_length:
-            chunk.append(word)
-        else:
-            chunks.append(" ".join(chunk))
-            chunk = [word]
-    
-    if chunk:
-        chunks.append(" ".join(chunk))
-    
+        if current_length + len(word) + 1 > max_length:
+            chunks.append(" ".join(current_chunk))
+            current_chunk = []
+            current_length = 0
+        current_chunk.append(word)
+        current_length += len(word) + 1
+
+    if current_chunk:
+        chunks.append(" ".join(current_chunk))
+
     return chunks
 
+
+# 🔹 Function: Call external TTS API
 def call_tts_api(text, voice):
-    """Sends text to TTS API and retries if it fails."""
-    filename = f"{SAVE_DIR}/audio_{uuid4().hex}.mp3"
-    
+    """Calls external TTS API and returns the saved MP3 filename."""
     try:
-        response = requests.post(TTS_API_URL, json={"text": text, "voice": voice})
-        response_data = response.json()
-        print("TTS API Response:", response_data)  # Log response
+        response = requests.post(
+            TTS_API_URL,
+            json={"text": text, "voice": voice},
+            headers={"Content-Type": "application/json"}
+        )
+        data = response.json()
         
-        if "audio_url" in response_data:
-            audio_url = response_data["audio_url"]
-            audio_data = requests.get(audio_url).content
-            
-            with open(filename, "wb") as f:
-                f.write(audio_data)
+        # ✅ Log response for debugging
+        print("TTS API Response:", data)
+        
+        if "audio_url" in data:
+            audio_url = data["audio_url"]
+            filename = f"audio_{uuid4().hex}.mp3"
+            audio_path = os.path.join(AUDIO_DIR, filename)
+
+            # ✅ Download the MP3 file
+            with requests.get(audio_url, stream=True) as r:
+                r.raise_for_status()
+                with open(audio_path, "wb") as f:
+                    for chunk in r.iter_content(chunk_size=8192):
+                        f.write(chunk)
             
-            return filename
+            return audio_path  # ✅ Return saved file path
         else:
-            print("Error: TTS API did not return an audio URL!")
+            print("Error: Invalid TTS API response", data)
             return None
-    
     except Exception as e:
-        print("Error calling TTS API:", str(e))
+        print("TTS API Error:", e)
         return None
 
-def merge_audio(files):
-    """Merges multiple MP3 files into one."""
-    output_file = f"{SAVE_DIR}/final_audio_{uuid4().hex}.mp3"
-    
-    try:
-        combined = AudioSegment.from_file(files[0])
-        
-        for file in files[1:]:
-            combined += AudioSegment.from_file(file)
-        
-        combined.export(output_file, format="mp3")
-        return output_file
-    except Exception as e:
-        print("Error merging audio files:", str(e))
+
+# 🔹 Function: Merge multiple MP3s into one
+def merge_audio(audio_files):
+    """Merges multiple MP3 files into one final MP3."""
+    if not audio_files:
         return None
 
+    final_audio = AudioSegment.empty()
+    for file in audio_files:
+        final_audio += AudioSegment.from_mp3(file)
+
+    final_filename = f"{AUDIO_DIR}/final_{uuid4().hex}.mp3"
+    final_audio.export(final_filename, format="mp3")
+    
+    return final_filename
+
+
+# ✅ API Endpoint: Generate Audio from Text
 @app.post("/generate-audio")
 def generate_audio(text: str = Query(...), voice: str = Query(...)):
     """Splits text, calls TTS API, merges audio, and returns the final MP3."""
@@ -94,15 +112,17 @@ def generate_audio(text: str = Query(...), voice: str = Query(...)):
     else:
         return {"error": "Failed to merge audio files"}
 
+
+# ✅ API Endpoint: Download the generated MP3
 @app.get("/download/{filename}")
 def download_file(filename: str):
-    """Serves the generated MP3 file."""
-    file_path = os.path.join(SAVE_DIR, filename)
+    """Allows users to download the generated MP3 file."""
+    file_path = os.path.join(AUDIO_DIR, filename)
     if os.path.exists(file_path):
-        return {"download_url": file_path}
-    else:
-        return {"error": "File not found"}
+        return {"download_url": f"https://hivecorp-manager1.hf.space/audio_files/{filename}"}
+    return {"error": "File not found"}
+
 
-# Ensure the app starts when running in Hugging Face Spaces
+# ✅ Ensure the app starts when running in Hugging Face Spaces
 if __name__ == "__main__":
     uvicorn.run(app, host="0.0.0.0", port=7860)