halimbahae commited on
Commit
e4eccab
·
verified ·
1 Parent(s): 5787cb6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +13 -13
app.py CHANGED
@@ -15,11 +15,9 @@ def generate_course(subject, history, lang, max_tokens, temperature, top_p):
15
  system_message = SYSTEM_PROMPT.get(lang, SYSTEM_PROMPT["en"]) # Sélection de la langue
16
 
17
  messages = [{"role": "system", "content": system_message}]
18
- for val in history:
19
- if val[0]:
20
- messages.append({"role": "user", "content": val[0]})
21
- if val[1]:
22
- messages.append({"role": "assistant", "content": val[1]})
23
 
24
  messages.append({"role": "user", "content": f"Crée un cours sur : {subject}"})
25
 
@@ -34,23 +32,25 @@ def generate_course(subject, history, lang, max_tokens, temperature, top_p):
34
  # Interface utilisateur avec Gradio
35
  with gr.Blocks(theme=gr.themes.Soft()) as demo:
36
  gr.Markdown("# 🎓 Teacher Assistant Chatbot")
37
-
38
  with gr.Row():
39
  subject_input = gr.Textbox(label="📌 Sujet du cours", placeholder="Ex: Apprentissage automatique")
40
  lang_select = gr.Dropdown(choices=["fr", "en"], value="fr", label="🌍 Langue")
41
-
42
- chat = gr.Chatbot()
43
-
44
  with gr.Row():
45
  max_tokens = gr.Slider(minimum=100, maximum=2048, value=512, step=1, label="📝 Max tokens")
46
  temperature = gr.Slider(minimum=0.1, maximum=2.0, value=0.7, step=0.1, label="🔥 Température")
47
  top_p = gr.Slider(minimum=0.1, maximum=1.0, value=0.95, step=0.05, label="🎯 Top-p")
48
 
49
  generate_button = gr.Button("🚀 Générer le cours")
50
-
51
- generate_button.click(generate_course,
52
- inputs=[subject_input, chat, lang_select, max_tokens, temperature, top_p],
53
- outputs=chat)
 
 
54
 
55
  # Lancer l'application
56
  if __name__ == "__main__":
 
15
  system_message = SYSTEM_PROMPT.get(lang, SYSTEM_PROMPT["en"]) # Sélection de la langue
16
 
17
  messages = [{"role": "system", "content": system_message}]
18
+
19
+ for message in history:
20
+ messages.append({"role": message["role"], "content": message["content"]})
 
 
21
 
22
  messages.append({"role": "user", "content": f"Crée un cours sur : {subject}"})
23
 
 
32
  # Interface utilisateur avec Gradio
33
  with gr.Blocks(theme=gr.themes.Soft()) as demo:
34
  gr.Markdown("# 🎓 Teacher Assistant Chatbot")
35
+
36
  with gr.Row():
37
  subject_input = gr.Textbox(label="📌 Sujet du cours", placeholder="Ex: Apprentissage automatique")
38
  lang_select = gr.Dropdown(choices=["fr", "en"], value="fr", label="🌍 Langue")
39
+
40
+ chat = gr.Chatbot(type="messages") # ✅ Fix : Utilisation du format messages
41
+
42
  with gr.Row():
43
  max_tokens = gr.Slider(minimum=100, maximum=2048, value=512, step=1, label="📝 Max tokens")
44
  temperature = gr.Slider(minimum=0.1, maximum=2.0, value=0.7, step=0.1, label="🔥 Température")
45
  top_p = gr.Slider(minimum=0.1, maximum=1.0, value=0.95, step=0.05, label="🎯 Top-p")
46
 
47
  generate_button = gr.Button("🚀 Générer le cours")
48
+
49
+ generate_button.click(
50
+ generate_course,
51
+ inputs=[subject_input, chat, lang_select, max_tokens, temperature, top_p],
52
+ outputs=chat
53
+ )
54
 
55
  # Lancer l'application
56
  if __name__ == "__main__":