Upload 4 files

main.py

@@ -1,5 +1,6 @@
 from fastapi import FastAPI, File, UploadFile
 from fastapi.responses import JSONResponse
+from fastapi.middleware.cors import CORSMiddleware
 from PIL import Image
 import io
 import json
@@ -13,6 +14,11 @@ app = FastAPI(
     """
 )
 
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins="*"
+)
+
 @app.post("/upload")
 async def upload_image(fields:str, model:str, file: UploadFile = File(...)):
     """
@@ -57,7 +63,7 @@ async def upload_image(fields:str, model:str, file: UploadFile = File(...)):
     except Exception as e:
         return JSONResponse(content={"error": str(e)}, status_code=400)
     
-@app.post("/list_models")
+@app.get("/models")
 async def list_models():
     """
     ### Endpoint Description:
@@ -66,4 +72,38 @@ async def list_models():
     #### Response:
     - A list of available models for text generation.
     """
-    return JSONResponse(content={"models": ["gpt-4o-mini", "gpt-4o", "deepseek-chat", "claude-3-5-sonnet-20241022", "llama_llm_d","llama_llm_o"]})
+
+    models = [
+        {
+            "id": "gpt-4o-mini",
+            "description": "Latest model by OpenAI prefered for accuracy",
+            "title": "GPT-4.o Mini",
+        },
+        {
+            "id": "gpt-4o",
+            "description": "Latest model by OpenAI prefered for accuracy",
+            "title": "GPT-4.o",
+        },
+        {
+            "id": "deepseek-chat",
+            "description": "Latest model by Deepseek prefered for speed",
+            "title": "DeepSeek Chat",
+        },
+        {
+            "id": "claude-3-5-sonnet-20241022",
+            "description": "Latest model by Google for both accuracy and speed",
+            "title": "Claude 3.5 Sonnet 20241022",
+        },
+        {
+            "id": "llama_llm_d",
+            "description": "Latest model by Facebook for both accuracy and speed",
+            "title": "Llama LLM D",
+        },
+        {
+            "id": "llama_llm_o",
+            "description": "Latest model by Facebook for both accuracy and speed",
+            "title": "Llama LLM O",
+        }
+    ]
+
+    return JSONResponse(content={"models": models})