Update main.py

main.py

@@ -36,7 +36,7 @@ class InputImage(BaseModel):
     cfg: int = 7
     seed: int = -1
 
-class RiattivaSpazio(BaseModel):
+class PostSpazio(BaseModel):
     nomeSpazio: str
     input: str = '' 
     api_name: str = "/chat"
@@ -78,6 +78,7 @@ def format_prompt(message, history):
     now = datetime.now().strftime("%Y-%m-%d %H:%M:%S.%f")
     prompt += f"[{now}] [INST] {message} [/INST]"
     return prompt
+
     
 #--------------------------------------------------- Generazione IMMAGINE ------------------------------------------------------
 @app.post("/Immagine")
@@ -111,15 +112,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):
+    
+#--------------------------------------------------- API PostSpazio ------------------------------------------------------
+@app.post("/PostSpazio")
+def generate_image(request: Request, input_data: PostSpazio):
     client = Client(input_data.nomeSpazio)
     result = client.predict(
     		input_data.input,	
     		api_name=input_data.api_name
     )
-    return {"response": "Spazio riattivato"}
+    return {"response": result}
 
 @app.get("/")
 def read_general():