Update app.py

app.py

@@ -14,6 +14,10 @@ if not os.path.exists(musl_path):
 import gradio as gr
 from huggingface_hub import hf_hub_download
 from llama_cpp import Llama
+import wikipedia
+
+# 1. Imposta Wikipedia Ufficialmente in Sardo (LSC/Logudorese/Campidanese)
+wikipedia.set_lang("sc")
 
 REPO_ID = "FPll/limba-mentor-llama3-gguf"
 FILENAME = "Meta-Llama-3.1-8B.Q4_K_M.gguf"
@@ -33,15 +37,41 @@ llm = Llama(
     verbose=False
 )
 
+def cerca_su_wikipedia(query):
+    """Cerca su Wikipedia in sardo e restituisce un piccolo riassunto"""
+    try:
+        # Pulisce la domanda per trovare meglio l'argomento (es. da "Chi è Grazia Deledda?" a "Grazia Deledda")
+        query_pulita = query.replace("Chie est", "").replace("Chi è", "").replace("Cos'è", "").replace("Ite est", "").strip()
+        
+        risultati = wikipedia.search(query_pulita, results=1)
+        if risultati:
+            titolo = risultati[0]
+            riassunto = wikipedia.summary(titolo, sentences=3) # Prende le prime 3 frasi
+            return f"[INFORMATZIONES CORTAS DAE WIKIPEDIA: {riassunto}]"
+        return ""
+    except Exception as e:
+        # Se la pagina non esiste o c'è un errore, proseguiamo senza
+        return ""
+
 def respond(message, history):
     # 1. IL SALVAVITA: Emette un segnale vuoto per impedire a Gradio di andare in crash (StopAsyncIteration)
     partial_message = ""
     yield partial_message 
     
     try:
-        # Estrazione sicura del testo nel caso Gradio invii un dizionario
         msg_text = message.get("text", "") if isinstance(message, dict) else str(message)
         
+        # --- IL TRUCCO RAG (Retrieval-Augmented Generation) ---
+        contesto_wiki = cerca_su_wikipedia(msg_text)
+        
+        if contesto_wiki:
+            # Se trova qualcosa, lo inietta di nascosto nella domanda
+            msg_arricchito = f"{contesto_wiki}\n\nRispondi alla seguente domanda basandoti ESCLUSIVAMENTE sulle informatziones de Wikipedia fornite qui sopra. Domanda: {msg_text}"
+        else:
+            # Se non trova niente, fa una domanda normale
+            msg_arricchito = msg_text
+        # ------------------------------------------------------
+        
         system_instruction = (
             "Sei Limba 2.0, creato da Francesco Palladino. Sei un modello di linguaggio che parla sardo (LSC) e italiano.\n"
             "REGOLE DA RISPETTARE:\n"
@@ -49,6 +79,7 @@ def respond(message, history):
             "2. Se l'utente chiede una formula o se nella risposta scrivi formule, usa il formato LaTeX racchiuso tra doppi simboli di dollaro ($$) su una riga separata, esempio: $$E=mc^2$$.\n"
             "3. Se l'utente ti saluta (es. 'Ciao', 'Buongiorno'), rispondi esclusivamente con: 'Salude! So Limba 2.0, comente ti potzo agiudare?' e fermati.\n"
             "4. Se l'utente fa una richiesta su un argomento specifico, rispondi alla richiesta in modo completo e accurato.\n"
+            "5. Se ricevi informazioni con il tag [INFORMATZIONES CORTAS DAE WIKIPEDIA], usale per rispondere senza inventare ASSOLUTAMENTE nulla. Traduci in sardo se necessario.\n"
         )
 
         few_shot = (