Update app.py

app.py

@@ -65,6 +65,25 @@ def process_image(image_file):
         }
     }
 
+def process_video(video_file):
+    video_bytes = video_file.read()
+    filetype = video_file.name.split(".")[-1].lower()
+    mime_map = {
+        "avi": "video/x-msvideo",
+        "mkv": "video/x-matroska",
+        "flv": "video/x-flv",
+        "wmv": "video/x-ms-wmv",
+        "mpeg": "video/mpeg"
+    }
+    mime_type = mime_map.get(filetype, video_file.type)
+    encoded_video = base64.b64encode(video_bytes).decode("utf-8")
+    return {
+        "inline_data": {
+            "mime_type": mime_type,
+            "data": encoded_video
+        }
+    }
+
 def export_conversation_to_pdf(conversation_history):
     pdf_path = "Conversation.pdf"
     doc = SimpleDocTemplate(pdf_path, pagesize=A4)
@@ -83,7 +102,7 @@ def export_conversation_to_pdf(conversation_history):
 # --------- Main App ---------
 def main():
     st.set_page_config(page_title="Gemini 2.5 Pro Chat Q&A", layout="wide")
-    st.title("📄 Chat with Your Documents leveraging Internet Using Gemini 2.5 Pro (Team 18))")
+    st.title("\U0001F4C4 Chat with Your Documents leveraging Internet Using Gemini 2.5 Pro (Team 18)")
 
     if "conversation" not in st.session_state:
         st.session_state.conversation = []
@@ -95,8 +114,12 @@ def main():
         st.session_state.chat_history = []
 
     uploaded_files = st.file_uploader(
-        "Upload files (.txt, .docx, .csv, .xlsx, .pptx, .pdf, .html, .htm, .tex, .jpg, .jpeg, .png):",
-        type=["txt", "docx", "csv", "xlsx", "pptx", "pdf", "html", "htm", "tex", "jpg", "jpeg", "png"],
+        "Upload files (.txt, .docx, .csv, .xlsx, .pptx, .pdf, .html, .htm, .tex, .jpg, .jpeg, .png, video formats):",
+        type=[
+            "txt", "docx", "csv", "xlsx", "pptx", "pdf", "html", "htm", "tex",
+            "jpg", "jpeg", "png",
+            "mp4", "webm", "mov", "avi", "mkv", "flv", "wmv", "mpeg"
+        ],
         accept_multiple_files=True
     )
 
@@ -122,19 +145,22 @@ def main():
                 elif filetype == "tex":
                     all_text_content += extract_text_from_tex(uploaded_file) + "\n"
                 elif filetype in ["jpg", "jpeg", "png"]:
-                    image_data = process_image(uploaded_file)
-                    st.session_state.image_data = image_data
+                    st.session_state.image_data = process_image(uploaded_file)
                     all_text_content += "Image uploaded and processed.\n"
                     st.image(uploaded_file, caption=uploaded_file.name, use_container_width=True)
+                elif filetype in ["mp4", "webm", "mov", "avi", "mkv", "flv", "wmv", "mpeg"]:
+                    st.session_state.video_data = process_video(uploaded_file)
+                    all_text_content += f"Video file '{uploaded_file.name}' uploaded and processed.\n"
+                    st.video(uploaded_file)
                 else:
                     st.error(f"Unsupported file format: {uploaded_file.name}")
             except Exception as e:
-                st.error(f"Failed to extract text from {uploaded_file.name}: {e}")
+                st.error(f"Failed to extract/process {uploaded_file.name}: {e}")
 
         st.session_state.documents_text = all_text_content
 
     if st.session_state.documents_text:
-        st.markdown("### 💬 Conversation")
+        st.markdown("### \U0001F4AC Conversation")
         for user_q, gemini_a in st.session_state.conversation:
             st.markdown(f"**You:** {user_q}")
             st.markdown(f"**Gemini:**\n\n{gemini_a}")
@@ -165,9 +191,11 @@ def main():
                             if "image_data" in st.session_state:
                                 content_blocks.append(st.session_state.image_data)
 
+                            if "video_data" in st.session_state:
+                                content_blocks.append(st.session_state.video_data)
+
                             content_blocks.append({"text": f"Question: {user_input}"})
 
-                            # ✅ Gemini with Internet Search
                             response = client.models.generate_content(
                                 model="gemini-2.5-pro-preview-06-05",
                                 contents=content_blocks,
@@ -175,7 +203,7 @@ def main():
                             )
 
                             st.session_state.conversation.append((user_input, response.text))
-                            st.success("💡 Answer:")
+                            st.success("\U0001F4A1 Answer:")
                             st.write(response.text)
 
                             st.session_state.chat_history.append({
@@ -187,7 +215,7 @@ def main():
             pdf_path = export_conversation_to_pdf(st.session_state.conversation)
             with open(pdf_path, "rb") as f:
                 st.download_button(
-                    label="📥 Export Conversation as PDF",
+                    label="\U0001F4E5 Export Conversation as PDF",
                     data=f,
                     file_name="Conversation.pdf",
                     mime="application/pdf"