FPll commited on
Commit
4d70ba5
·
verified ·
1 Parent(s): 82bdb96

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -6
app.py CHANGED
@@ -41,34 +41,38 @@ llm = Llama(
41
  def cerca_su_wikipedia(query):
42
  """Estrazione entità via API remota e ricerca (Sardo -> Italiano)"""
43
  try:
44
- # Prompt per il modello remoto: zero RAM locale
45
- prompt_estrazione = f"Estrai solo l'argomento principale da cercare su Wikipedia da questa domanda. Rispondi SOLO con la parola chiave, senza punteggiatura. Domanda: '{query}'"
46
 
47
- # Chiamata all'API gratuita (usiamo un modello veloce)
48
  query_pulita = hf_client.text_generation(
49
  prompt_estrazione,
50
  model="HuggingFaceH4/zephyr-7b-beta",
51
  max_new_tokens=10
52
  ).strip()
 
53
 
54
  if not query_pulita:
55
  return ""
56
 
57
- # 1. Tenta in sardo
58
  wikipedia.set_lang("sc")
59
  risultati = wikipedia.search(query_pulita, results=1)
 
60
 
61
- # 2. Fallback in italiano
62
  if not risultati:
63
  wikipedia.set_lang("it")
64
  risultati = wikipedia.search(query_pulita, results=1)
 
65
 
66
  if risultati:
67
  titolo = risultati[0]
68
  riassunto = wikipedia.summary(titolo, sentences=3)
 
69
  return f"[INFORMATZIONES DAE WIKIPEDIA: {riassunto}]"
 
 
70
  return ""
71
  except Exception as e:
 
72
  return ""
73
 
74
  def respond(message, history):
@@ -188,7 +192,7 @@ with gr.Blocks() as demo:
188
  examples=["Iscrie unu script in Python pro lèghere unu file CSV.", "Ispiegami su teorema de Pitagora.", "Traduci in sardo: L'algoritmo di machine learning sta analizzando i dati."],
189
  cache_examples=False,
190
  chatbot=gr.Chatbot(
191
- avatar_images=[None, "logo.png"],
192
  height=500
193
  )
194
  )
 
41
  def cerca_su_wikipedia(query):
42
  """Estrazione entità via API remota e ricerca (Sardo -> Italiano)"""
43
  try:
44
+ print(f"\n--- RAG START --- Query utente: '{query}'")
45
+ prompt_estrazione = f"Estrai solo l'argomento principale da cercare su Wikipedia da questa domanda. Rispondi SOLO con l'argomento, senza punteggiatura. Domanda: '{query}'"
46
 
 
47
  query_pulita = hf_client.text_generation(
48
  prompt_estrazione,
49
  model="HuggingFaceH4/zephyr-7b-beta",
50
  max_new_tokens=10
51
  ).strip()
52
+ print(f"--- RAG ESTRAZIONE --- Zephyr ha estratto: '{query_pulita}'")
53
 
54
  if not query_pulita:
55
  return ""
56
 
 
57
  wikipedia.set_lang("sc")
58
  risultati = wikipedia.search(query_pulita, results=1)
59
+ print(f"--- RAG SEARCH SC --- Risultati in sardo: {risultati}")
60
 
 
61
  if not risultati:
62
  wikipedia.set_lang("it")
63
  risultati = wikipedia.search(query_pulita, results=1)
64
+ print(f"--- RAG SEARCH IT --- Risultati in italiano: {risultati}")
65
 
66
  if risultati:
67
  titolo = risultati[0]
68
  riassunto = wikipedia.summary(titolo, sentences=3)
69
+ print(f"--- RAG SUCCESSO --- Testo estratto da: {titolo}")
70
  return f"[INFORMATZIONES DAE WIKIPEDIA: {riassunto}]"
71
+
72
+ print("--- RAG FALLITO --- Nessun risultato trovato.")
73
  return ""
74
  except Exception as e:
75
+ print(f"--- RAG ERRORE --- {str(e)}")
76
  return ""
77
 
78
  def respond(message, history):
 
192
  examples=["Iscrie unu script in Python pro lèghere unu file CSV.", "Ispiegami su teorema de Pitagora.", "Traduci in sardo: L'algoritmo di machine learning sta analizzando i dati."],
193
  cache_examples=False,
194
  chatbot=gr.Chatbot(
195
+ avatar_images=[None, "https://huggingface.co/spaces/FPll/Limba_2.0/resolve/main/logo.png"],
196
  height=500
197
  )
198
  )