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")
|