Update main.py

main.py

@@ -36,6 +36,11 @@ class InputImage(BaseModel):
     cfg: int = 7
     seed: int = -1
 
+class RiattivaSpazio(BaseModel):
+    nomeSpazio: str
+    input: str = '' 
+    api_name: str = "/chat"
+    
 #--------------------------------------------------- Generazione TESTO ------------------------------------------------------
 @app.post("/Genera")
 def read_root(request: Request, input_data: InputData):
@@ -106,6 +111,16 @@ def generate_image(request: Request, input_data: InputImage):
                 return {"error": "Errore interno del server persistente"}
     return {"error": "Numero massimo di tentativi raggiunto"}
 
+#--------------------------------------------------- API Spaces (per riattivarli) ------------------------------------------------------
+@app.post("/RiattivaSpazio")
+def generate_image(request: Request, input_data: RiattivaSpazio):
+    client = Client(input_data.nomeSpazio)
+    result = client.predict(
+    		input_data.input,	
+    		api_name=input_data.api_name
+    )
+    return {"response": "Spazio riattivato"}
+
 @app.get("/")
 def read_general():
     return {"response": "Benvenuto. Per maggiori info: https://matteoscript-fastapi.hf.space/docs"}