Update app.py

app.py

@@ -170,16 +170,17 @@ def capture_video():
     """
     html(video_recorder_html, height=400)
 
+# --------- Main App ---------
 # --------- Main App ---------
 def main():
     st.set_page_config(page_title="Gemini 2.0 Flash Chat Q&A", layout="wide")
     st.title("\U0001F4C4 Team 18 Comprehensive Prototype Using Gemini 2.0 Flash")
 
-    # Ensure that session state variables are correctly initialized
+    # Initialize session state variables
     if "conversation" not in st.session_state:
         st.session_state.conversation = []
     if "documents_text" not in st.session_state:
-        st.session_state.documents_text = ""  # Initialize as an empty string, not a list
+        st.session_state.documents_text = ""  # Initialize as an empty string
     if "chat_active" not in st.session_state:
         st.session_state.chat_active = True
     if "chat_history" not in st.session_state:
@@ -195,6 +196,7 @@ def main():
         accept_multiple_files=True
     )
 
+    # Handle the files uploaded
     if uploaded_files:
         all_text_content = ""
         for uploaded_file in uploaded_files:
@@ -283,28 +285,35 @@ def main():
                             if st.session_state.documents_text.strip():
                                 content_blocks.append({"text": f"Context:\n{st.session_state.documents_text}"})
 
-                            if "image_data" in st.session_state:
-                                content_blocks.append(st.session_state.image_data)
+                            # Send image data if available
+                            if "image_data" in st.session_state and st.session_state.image_data:
+                                content_blocks.append({"image": st.session_state.image_data})
 
-                            if "video_data" in st.session_state:
-                                content_blocks.append(st.session_state.video_data)
+                            # Send video data if available
+                            if "video_data" in st.session_state and st.session_state.video_data:
+                                content_blocks.append({"video": st.session_state.video_data})
 
                             content_blocks.append({"text": f"Question: {user_input}"})
 
-                            response = client.models.generate_content(
-                                model="gemini-2.0-flash",
-                                contents=content_blocks,
-                                config={"tools": [{"google_search": {}}]}
-                            )
+                            try:
+                                # Ensure we are sending the data in the correct format
+                                response = client.models.generate_content(
+                                    model="gemini-2.0-flash",
+                                    contents=content_blocks,
+                                    config={"tools": [{"google_search": {}}]}
+                                )
 
-                            st.session_state.conversation.append((user_input, response.text))
-                            st.success("\U0001F4A1 Answer:")
-                            st.write(response.text)
+                                st.session_state.conversation.append((user_input, response.text))
+                                st.success("\U0001F4A1 Answer:")
+                                st.write(response.text)
 
-                            st.session_state.chat_history.append({
-                                "question": user_input,
-                                "answer": response.text
-                            })
+                                st.session_state.chat_history.append({
+                                    "question": user_input,
+                                    "answer": response.text
+                                })
+
+                            except Exception as e:
+                                st.error(f"Error with Gemini API request: {e}")
 
         if st.session_state.conversation:
             pdf_path = export_conversation_to_pdf(st.session_state.conversation)
@@ -317,4 +326,6 @@ def main():
                 )
 
 if __name__ == "__main__":
-    main()
+    main()
+
+