Update app.py

app.py

@@ -176,7 +176,20 @@ def main():
                         st.success("Chat ended. Reload to start again.")
                     else:
                         with st.spinner("Gemini is thinking..."):
-                            content_blocks = []
+                            # -------- System context for Gemini --------
+                            system_context = {
+                                "role": "user",
+                                "text": (
+                                    "You are part of an intelligent driver distraction detection system. "
+                                    "Your role is to analyze documents, images, and videos to help identify and classify driver states, "
+                                    "including: drowsy, alert, yawning, microsleep, eyes/lips open or closed, and other distractions. "
+                                    "The user may ask for timestamps of events in video, technical explanations from code/docs, "
+                                    "or structured summaries. Prioritize your answers for this use case and act as a helpful assistant "
+                                    "within this domain."
+                                )
+                            }
+
+                            content_blocks = [system_context]
 
                             if st.session_state.conversation:
                                 history_text = "\n\n".join(