Update app.py

app.py

@@ -48,7 +48,7 @@ except Exception as e:
 
 # Inizializza il client di Hugging Face
 try:
-    client = InferenceClient(HF_MODEL, token=HF_API_KEY)
+    client = InferenceClient(model=HF_MODEL, token=HF_API_KEY)
     logger.info("InferenceClient inizializzato correttamente.")
 except Exception as e:
     logger.error(f"Errore nell'inizializzazione di InferenceClient: {e}")
@@ -234,6 +234,8 @@ async def call_hf_model(prompt: str, temperature: float = 0.5, max_tokens: int =
             stream=stream
         )
 
+        logger.debug(f"Risposta completa dal modello: {response}")
+
         if stream:
             # Gestisci lo stream
             generated_text = ""
@@ -244,15 +246,22 @@ async def call_hf_model(prompt: str, temperature: float = 0.5, max_tokens: int =
             return generated_text.strip()
         else:
             # Risposta non in streaming
-            if isinstance(response, list) and len(response) > 0 and "generated_text" in response[0]:
-                raw = response[0]["generated_text"]
-            elif "generated_text" in response:
-                raw = response["generated_text"]
+            # Logga la risposta completa per capire la struttura
+            logger.debug(f"Risposta completa: {response}")
+
+            # Adatta la logica di estrazione del testo generato
+            if isinstance(response, dict):
+                if 'choices' in response and len(response['choices']) > 0:
+                    generated_text = response['choices'][0].get('message', {}).get('content', '')
+                else:
+                    raise ValueError("Risposta non contiene 'choices' o 'message'.")
+            elif isinstance(response, list) and len(response) > 0:
+                generated_text = response[0].get('message', {}).get('content', '')
             else:
-                raise ValueError("Nessun campo 'generated_text' nella risposta.")
+                raise ValueError("Struttura della risposta non riconosciuta.")
 
             # Forza la risposta su una singola linea se multilinea
-            single_line = " ".join(raw.splitlines())
+            single_line = " ".join(generated_text.splitlines())
             logger.debug(f"Risposta HF single-line: {single_line}")
             return single_line.strip()
     except Exception as e:
@@ -413,12 +422,14 @@ async def generate_response(req: QueryRequest):
         "sparql_results": row_list,
         "explanation": explanation
     }
+
 @app.post("/prova")
 async def prova(req: QueryRequest):
     return {
         "type": "NATURAL",
-        "response":"Questa è una prova di richiesta"
+        "response": "Questa è una prova di richiesta"
     }
+
 @app.get("/")
 def home():
     return {"message": "Assistente Museale con supporto SPARQL."}