awd

app.py

@@ -15,7 +15,7 @@ models = {
     },
 }
 
-# Chat Function
+# Chat Function with Context
 def chat_with_model(model_choice, user_message, chat_history, file=None):
     if model_choice == "PDF Summarizer (T5)" and file is not None:
         pdf_text = extract_text_from_pdf(file)
@@ -27,11 +27,16 @@ def chat_with_model(model_choice, user_message, chat_history, file=None):
     model_info = models[model_choice]
     client = model_info["client"]
 
-    # Prepare messages for the InferenceClient
-    messages = [
-        {"role": "system", "content": "You are a helpful assistant."},
-        {"role": "user", "content": user_message}
-    ]
+    # Prepare messages for the InferenceClient including chat history
+    messages = [{"role": "system", "content": "You are a helpful assistant."}]
+    
+    # Add previous conversation to the messages
+    for user_msg, bot_msg in chat_history:
+        messages.append({"role": "user", "content": user_msg})
+        messages.append({"role": "assistant", "content": bot_msg})
+    
+    # Add the current user message
+    messages.append({"role": "user", "content": user_message})
 
     # Generate Response
     response = ""