Upload 3 files

Dockerfile

@@ -0,0 +1,20 @@
+# Use an official Python runtime as a base image
+FROM python:3.9-slim
+
+# Set the working directory in the container
+WORKDIR /app
+
+# Copy the requirements file into the container
+COPY requirements.txt .
+
+# Install the dependencies
+RUN pip install --no-cache-dir -r requirements.txt
+
+# Copy the FastAPI app into the container
+COPY . .
+
+# Expose port 7860 (the default port for Hugging Face Spaces)
+EXPOSE 7860
+
+# Command to run the FastAPI app with Uvicorn
+CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]

main.py

@@ -0,0 +1,20 @@
+from fastapi import FastAPI, HTTPException, Response
+from pydantic import BaseModel
+from plantuml import PlantUML
+
+app = FastAPI()
+
+class PlantUMLRequest(BaseModel):
+    script: str
+
+# Initialize the PlantUML server URL
+plantuml_server = PlantUML(url="http://www.plantuml.com/plantuml/png/")
+
+@app.post("/generate-image/")
+async def generate_image(request: PlantUMLRequest):
+    try:
+        # Use the `processes` function to get the raw PNG data
+        image_data = plantuml_server.processes(request.script)
+        return Response(content=image_data, media_type="image/png")
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))