Update app.py

app.py

@@ -178,6 +178,20 @@ def main():
                         with st.spinner("Gemini is thinking..."):
                             content_blocks = []
 
+                            # System prompt inserted first here:
+                            content_blocks.append({
+                                "text": (
+                                    "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. "
+                                    "Additionally, your classification may rely on the State Farm Distracted Driver Detection dataset, "
+                                    "which defines 10 distraction categories such as texting, phone use, reaching behind, adjusting the radio, "
+                                    "and safe driving. Use this labeling scheme when interpreting relevant images."
+                                )
+                            })
+
                             if st.session_state.conversation:
                                 history_text = "\n\n".join(
                                     f"Q: {entry[0]}\nA: {entry[1]}"
@@ -222,4 +236,4 @@ def main():
                 )
 
 if __name__ == "__main__":
-    main()
+    main()