Update app.py
Browse files
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 |
-
# 🔄
|
35 |
for message in history:
|
36 |
if isinstance(message, dict) and "role" in message and "content" in message:
|
37 |
messages.append(message)
|
38 |
|
39 |
-
# 📄
|
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]}..."}) #
|
43 |
|
44 |
-
#
|
45 |
messages.append({"role": "user", "content": f"Crée un cours sur : {subject}"})
|
46 |
|
47 |
-
# 🔥
|
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")
|