Pamudu13 commited on
Commit
5e58171
·
verified ·
1 Parent(s): 2d35ce4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -7
app.py CHANGED
@@ -79,14 +79,17 @@ def generate_response(
79
 
80
  context = search_relevant_text(message) # Get relevant content from PDF
81
 
82
- messages = [{"role": "system", "content": system_message}]
83
- for user_msg, bot_msg in history:
84
- if user_msg:
85
- messages.append({"role": "user", "content": user_msg})
86
- if bot_msg:
 
 
 
87
  messages.append({"role": "assistant", "content": bot_msg})
88
-
89
- messages.append({"role": "user", "content": f"Context: {context}\nQuestion: {message}"})
90
 
91
  try:
92
  response = ""
 
79
 
80
  context = search_relevant_text(message) # Get relevant content from PDF
81
 
82
+ # Start with the system message in the first user message
83
+ messages = []
84
+ first_msg = f"{system_message}\n\nContext: {context}\nQuestion: {message}"
85
+ messages.append({"role": "user", "content": first_msg})
86
+
87
+ # Add conversation history ensuring alternating pattern
88
+ for i, (user_msg, bot_msg) in enumerate(history):
89
+ if bot_msg: # Add assistant's previous response
90
  messages.append({"role": "assistant", "content": bot_msg})
91
+ if user_msg: # Add user's previous message
92
+ messages.append({"role": "user", "content": user_msg})
93
 
94
  try:
95
  response = ""