Update app.py

app.py

@@ -1,15 +1,21 @@
 from fastapi import FastAPI, UploadFile, File, HTTPException
+from fastapi.responses import Response
 import yt_dlp
 import whisper
 import tempfile
 import os
+import io
+from diffusers import StableDiffusionPipeline
 
 app = FastAPI(title="Silent Tech Utils API")
 
-# Load the AI Voice model into RAM (Using 'base' so it's super fast on a CPU)
-print("Loading Whisper AI...")
-model = whisper.load_model("base")
-print("Whisper ready!")
+print("Loading Whisper Voice AI...")
+voice_model = whisper.load_model("base")
+
+print("Loading Uncensored Image AI...")
+# safety_checker=None completely disables the NSFW and violence filters!
+image_model = StableDiffusionPipeline.from_pretrained("prompthero/openjourney", safety_checker=None)
+image_model.to("cpu")
 
 @app.get("/")
 def read_root():
@@ -18,21 +24,14 @@ def read_root():
 @app.get("/api/download")
 def download_media(url: str):
     """Bypasses protections and gets the direct raw MP4/MP3 link."""
-    ydl_opts = {
-        'format': 'best',
-        'quiet': True,
-        'no_warnings': True,
-        'skip_download': True # We just want the URL to give to the WhatsApp bot
-    }
+    ydl_opts = {'format': 'best', 'quiet': True, 'no_warnings': True, 'skip_download': True}
     try:
         with yt_dlp.YoutubeDL(ydl_opts) as ydl:
             info = ydl.extract_info(url, download=False)
             return {
                 "success": True,
                 "title": info.get('title'),
-                "duration": info.get('duration'),
-                "thumbnail": info.get('thumbnail'),
-                "download_url": info.get('url') # The direct raw video/audio link!
+                "download_url": info.get('url') # The direct raw link!
             }
     except Exception as e:
         raise HTTPException(status_code=400, detail=str(e))
@@ -41,17 +40,28 @@ def download_media(url: str):
 async def transcribe_audio(file: UploadFile = File(...)):
     """Converts WhatsApp voice notes to text."""
     try:
-        # Save the uploaded WhatsApp audio temporarily
         with tempfile.NamedTemporaryFile(delete=False, suffix=".ogg") as temp_audio:
             temp_audio.write(await file.read())
             temp_audio_path = temp_audio.name
-        
-        # Whisper AI transcribes it to text
-        result = model.transcribe(temp_audio_path)
-        
-        # Clean up
+            
+        result = voice_model.transcribe(temp_audio_path)
         os.remove(temp_audio_path)
-        
         return {"success": True, "text": result["text"].strip()}
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))
+
+@app.get("/api/image")
+def generate_image(prompt: str):
+    """Generates an uncensored image and returns it directly as a PNG."""
+    try:
+        # Generate the image (15 steps is faster for CPU)
+        image = image_model(prompt, num_inference_steps=15).images[0]
+        
+        # Convert it to a raw PNG image file
+        img_bytes = io.BytesIO()
+        image.save(img_bytes, format="PNG")
+        
+        # Send the actual image picture directly to the bot, not just text!
+        return Response(content=img_bytes.getvalue(), media_type="image/png")
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))