Update main.py

main.py

@@ -15,6 +15,8 @@ import aiohttp
 import asyncio
 import json
 from types import SimpleNamespace
+from io import BytesIO
+from PIL import Image
 
 #--------------------------------------------------- Definizione Server FAST API ------------------------------------------------------
 app = FastAPI()
@@ -500,6 +502,27 @@ def generate_image(request: Request, input_data: InputImage):
                 return {"error": "Errore interno del server persistente!"}
     return {"error": "Numero massimo di tentativi raggiunto"}
 
+#--------------------------------------------------- IMAGE TO TEXT ------------------------------------------------------
+class InputImageToText(BaseModel): 
+    base64: str
+    input: str = ''
+
+def base64_in_immagine(dati_base64):
+    immagine = base64.b64decode(dati_base64)
+    immagine_pil = Image.open(BytesIO(immagine))
+    nome_file = "img.jpg"
+    immagine_pil.save(nome_file)
+    
+@app.post("/Image_To_Text")
+def image_to_text(request: Request, input_data: InputImageToText):
+    base64_in_immagine(InputImageToText.base64)
+    client = Client("vikhyatk/moondream2")
+    result = client.predict(
+    		"img.jpg",	
+    		"Describe Image",	
+    		api_name="/answer_question_1"
+    )
+    return {"response": result}
     
 #--------------------------------------------------- API PostSpazio ------------------------------------------------------
 @app.post("/PostSpazio")