Create app.py

app.py

@@ -0,0 +1,73 @@
+from fastapi import FastAPI, HTTPException
+from fastapi.responses import StreamingResponse
+from fastapi.middleware.cors import CORSMiddleware
+import io
+import wave
+import math
+import struct
+
+# Initialize the FastAPI app
+app = FastAPI()
+
+# IMPORTANT: Add CORS middleware to allow requests from your frontend
+# This is crucial for connecting the two parts.
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],  # Allows all origins
+    allow_credentials=True,
+    allow_methods=["*"],  # Allows all methods
+    allow_headers=["*"],  # Allows all headers
+)
+
+def create_mock_wav_in_memory(prompt: str) -> io.BytesIO:
+    """Generates a 1-second, 440Hz sine wave WAV file in memory."""
+    sample_rate = 44100
+    duration_seconds = 1
+    frequency = 440.0  # A4 note
+    num_samples = duration_seconds * sample_rate
+    
+    # Use io.BytesIO to build the WAV file in memory
+    wav_file_in_memory = io.BytesIO()
+    
+    with wave.open(wav_file_in_memory, 'wb') as w:
+        w.setnchannels(1)  # Mono
+        w.setsampwidth(2)  # 16-bit PCM
+        w.setframerate(sample_rate)
+        
+        for i in range(num_samples):
+            # Calculate the sample value for the sine wave
+            value = int(32767.0 * math.sin(2 * math.pi * frequency * i / sample_rate))
+            # Pack the value as a 16-bit signed integer
+            data = struct.pack('<h', value)
+            w.writeframesraw(data)
+
+    # Go back to the beginning of the in-memory file so it can be read
+    wav_file_in_memory.seek(0)
+    return wav_file_in_memory
+
+@app.post("/generate-audio")
+async def generate_audio_endpoint(payload: dict):
+    """
+    This endpoint receives a text prompt and returns a generated audio file.
+    """
+    prompt = payload.get("prompt")
+    if not prompt:
+        raise HTTPException(status_code=400, detail="A 'prompt' is required in the request body.")
+
+    # Generate the mock audio data
+    audio_data_stream = create_mock_wav_in_memory(prompt)
+    
+    # Create a safe filename from the prompt
+    safe_filename = "".join(c for c in prompt if c.isalnum() or c in (' ', '_')).rstrip()[:50] + ".wav"
+    
+    # Return the in-memory WAV file as a streaming response
+    return StreamingResponse(
+        audio_data_stream,
+        media_type="audio/wav",
+        headers={"Content-Disposition": f"attachment; filename=\"{safe_filename}\""}
+    )
+
+@app.get("/")
+def read_root():
+    """A simple root endpoint to confirm the API is running."""
+    return {"message": "Mock Audio Generation API is running."}