added get captcha endpoint

app/main.py

@@ -1,23 +1,46 @@
-from fastapi import FastAPI
+import os
+import random
+from pathlib import Path
+from fastapi import FastAPI, HTTPException
 from fastapi.middleware.cors import CORSMiddleware
 
 # Initialize the FastAPI app
 app = FastAPI()
 
 # --- CORS Middleware ---
-# This is crucial for allowing your frontend to communicate with this backend.
-# origins = ["*"] allows all origins, which is fine for this project.
-# For a production application, you might want to restrict this to your frontend's domain.
 app.add_middleware(
     CORSMiddleware,
-    allow_origins=["*"],  # Allows all origins
+    allow_origins=["*"],
     allow_credentials=True,
-    allow_methods=["*"],  # Allows all methods (GET, POST, etc.)
-    allow_headers=["*"],  # Allows all headers
+    allow_methods=["*"],
+    allow_headers=["*"],
 )
 
+# --- Constants ---
+# Define the path to the directory containing captcha images
+IMAGE_DIR = Path("static/images")
+
+
 # --- API Endpoints ---
 @app.get("/")
 async def read_root():
     """A simple root endpoint to check if the API is running."""
-    return {"message": "Welcome to the Captcha Solver API!"}
+    return {"message": "Welcome to the Captcha Solver API!"}
+
+
+@app.get("/get_captcha")
+async def get_captcha():
+    """
+    Returns the filename of a random captcha image from the static/images directory.
+    """
+    if not IMAGE_DIR.is_dir():
+        raise HTTPException(status_code=500, detail="Image directory not found on server.")
+
+    image_files = [f for f in os.listdir(IMAGE_DIR) if f.endswith(('.png'))]
+
+    if not image_files:
+        raise HTTPException(status_code=404, detail="No captcha images found.")
+
+    random_image_filename = random.choice(image_files)
+
+    return {"filename": random_image_filename}