Elouarn commited on
Commit
75fbbbc
·
verified ·
1 Parent(s): 26d1406

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -3
app.py CHANGED
@@ -3,8 +3,9 @@ import gradio as gr
3
 
4
  client = InferenceClient("mistralai/Mixtral-8x7B-Instruct-v0.1")
5
 
6
- # Variable globale pour stocker le contexte
7
- # global_context = {"preprompt": "Bienvenue dans ma taverne!", "hidden_context": "tu es le personnage dun escape game sur la resolution d un meutre dans l univers des cathares au 14 eme siecle dans les Pyrénées, tu es le gérant d une taverne qui est cathare dans le village de montaillou, parle que en français, assez rustre mais qui demeure accueillant tant que son interlocuteur n est pas catholique ou du moins quelqu un avec une influence tel qu un noble, tu n es jamais hors contexte par rapport à ton époque, tes connaissances se limite à ta situation sociale et ton époque, il faut que tu sois immersif pour la personne qui va te parler, et tu dois cacher ton implication dans le meutre si on te pose des questions à ce sujet tu dois tourner la discussion à ton avantage et le dévier du meutre"}
 
8
 
9
  def format_prompt(message, history):
10
  prompt = "<s>"
@@ -44,6 +45,8 @@ def generate(prompt, history, temperature=0.2, max_new_tokens=256, top_p=0.95, r
44
  # Exclure le contexte caché de la réponse retournée par le modèle
45
  response_text = response.token.text.replace(global_context['hidden_context'], "")
46
  output += response_text
 
 
47
  yield output
48
  return output
49
 
@@ -54,7 +57,9 @@ demo = gr.ChatInterface(fn=generate,
54
  chatbot=mychatbot,
55
  title="Tavernier 🍺",
56
  retry_btn=None,
57
- undo_btn=None
 
 
58
  )
59
 
60
  # Lancer l'interface
 
3
 
4
  client = InferenceClient("mistralai/Mixtral-8x7B-Instruct-v0.1")
5
 
6
+ # Variable globale pour stocker le contexte et l'historique de la conversation
7
+ global_context = {"preprompt": "Bienvenue dans ma taverne!", "hidden_context": "tu es le personnage dun escape game sur la resolution d un meutre dans l univers des cathares au 14 eme siecle dans les Pyrénées, tu es le gérant d une taverne qui est cathare dans le village de montaillou, parle que en français, assez rustre mais qui demeure accueillant tant que son interlocuteur n est pas catholique ou du moins quelqu un avec une influence tel qu un noble, tu n es jamais hors contexte par rapport à ton époque, tes connaissances se limite à ta situation sociale et ton époque, il faut que tu sois immersif pour la personne qui va te parler, et tu dois cacher ton implication dans le meutre si on te pose des questions à ce sujet tu dois tourner la discussion à ton avantage et le dévier du meutre"}
8
+ conversation_history = []
9
 
10
  def format_prompt(message, history):
11
  prompt = "<s>"
 
45
  # Exclure le contexte caché de la réponse retournée par le modèle
46
  response_text = response.token.text.replace(global_context['hidden_context'], "")
47
  output += response_text
48
+ # Mettre à jour l'historique de la conversation
49
+ conversation_history.append((formatted_prompt, response_text))
50
  yield output
51
  return output
52
 
 
57
  chatbot=mychatbot,
58
  title="Tavernier 🍺",
59
  retry_btn=None,
60
+ undo_btn=None,
61
+ inputs=["text"],
62
+ outputs=["text"]
63
  )
64
 
65
  # Lancer l'interface