halimbahae commited on
Commit
75a7bc2
·
verified ·
1 Parent(s): 21a022f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -5
app.py CHANGED
@@ -29,22 +29,23 @@ def extract_text_from_pdf(pdf_path):
29
  def generate_response(subject, history, lang, pdf_path, max_tokens, temperature, top_p):
30
  system_message = SYSTEM_PROMPT.get(lang, SYSTEM_PROMPT["en"]) # Sélection de la langue
31
 
 
32
  messages = [{"role": "system", "content": system_message}]
33
 
34
- # 🔄 Correction du format de l'historique
35
  for message in history:
36
  if isinstance(message, dict) and "role" in message and "content" in message:
37
  messages.append(message)
38
 
39
- # 📄 Ajouter le contenu du PDF s'il y en a un
40
  if pdf_path:
41
  pdf_text = extract_text_from_pdf(pdf_path)
42
- messages.append({"role": "user", "content": f"Voici un document PDF pertinent : {pdf_text[:1000]}..."}) # On limite à 1000 caractères pour éviter la surcharge
43
 
44
- # Ajouter la demande de l'utilisateur
45
  messages.append({"role": "user", "content": f"Crée un cours sur : {subject}"})
46
 
47
- # 🔥 Génération de la réponse en streaming
48
  response = ""
49
  for message in client.chat_completion(
50
  messages, max_tokens=max_tokens, stream=True, temperature=temperature, top_p=top_p
@@ -53,6 +54,7 @@ def generate_response(subject, history, lang, pdf_path, max_tokens, temperature,
53
  response += token
54
  yield response
55
 
 
56
  # 🎨 Interface utilisateur Gradio
57
  with gr.Blocks(theme=gr.themes.Soft()) as demo:
58
  gr.Markdown("# 🎓 Teacher Assistant Chatbot avec PDF RAG")
 
29
  def generate_response(subject, history, lang, pdf_path, max_tokens, temperature, top_p):
30
  system_message = SYSTEM_PROMPT.get(lang, SYSTEM_PROMPT["en"]) # Sélection de la langue
31
 
32
+ # Initialize messages with the system message
33
  messages = [{"role": "system", "content": system_message}]
34
 
35
+ # 🔄 Correct format for history messages
36
  for message in history:
37
  if isinstance(message, dict) and "role" in message and "content" in message:
38
  messages.append(message)
39
 
40
+ # 📄 Add PDF content if available
41
  if pdf_path:
42
  pdf_text = extract_text_from_pdf(pdf_path)
43
+ messages.append({"role": "user", "content": f"Voici un document PDF pertinent : {pdf_text[:1000]}..."}) # Limit to first 1000 characters
44
 
45
+ # Add user's request to create a course
46
  messages.append({"role": "user", "content": f"Crée un cours sur : {subject}"})
47
 
48
+ # 🔥 Stream response from HuggingFace model
49
  response = ""
50
  for message in client.chat_completion(
51
  messages, max_tokens=max_tokens, stream=True, temperature=temperature, top_p=top_p
 
54
  response += token
55
  yield response
56
 
57
+
58
  # 🎨 Interface utilisateur Gradio
59
  with gr.Blocks(theme=gr.themes.Soft()) as demo:
60
  gr.Markdown("# 🎓 Teacher Assistant Chatbot avec PDF RAG")