Update app.py

app.py

@@ -34,26 +34,24 @@ def chat(message, history):
     try:
         pdf_content = get_pdf_content()
         
+        # Prepare history with previous messages
+        history_messages = [{"role": "user", "parts": [msg[0]]} for msg in history]
+        
+        # Create base message with system prompt and PDF if available
         if pdf_content:
-            messages = [
+            base_message = [
                 {"role": "user", "parts": [
                     system_prompt,
-                    types.Part.from_bytes(
-                        data=pdf_content,
-                        mime_type='application/pdf'
-                    ),
+                    types.Part.from_bytes(data=pdf_content, mime_type='application/pdf'),
                     "Use the PDF content to provide detailed answers about the course."
-                ]},
-                *[{"role": "user", "parts": [msg[0]]} for msg in history],
-                {"role": "user", "parts": [message]}
+                ]}
             ]
         else:
-            messages = [
-                {"role": "user", "parts": [system_prompt]},
-                *[{"role": "user", "parts": [msg[0]]} for msg in history],
-                {"role": "user", "parts": [message]}
-            ]
-
+            base_message = [{"role": "user", "parts": [system_prompt]}]
+        
+        # Combine all messages
+        messages = base_message + history_messages + [{"role": "user", "parts": [message]}]
+        
         response = model.generate_content(messages)
         return response.text
     except Exception as e: