Create app.py

app.py

@@ -0,0 +1,81 @@
+import streamlit as st
+import base64
+import io
+import google.genai as genai  # Import SDK của google-genai
+
+# Cấu hình API key và model cho Gemini
+API_KEY = "AIzaSyAD3PeWI8ie3LNDaMqoIoIf6HyfDT6Bfgs"
+MODEL_ID = "gemini-2.0-pro-exp-02-05"
+genai.configure(api_key=API_KEY)
+
+st.title("Gemini 2.0 Pro Chatbot")
+st.write("Chào mừng đến với chatbot Gemini sử dụng google-genai SDK!")
+
+# Nhập nội dung chat
+user_prompt = st.text_input("Nhập nội dung chat:")
+
+# File uploader hỗ trợ nhiều loại file: hình ảnh, video, audio, PDF
+uploaded_files = st.file_uploader(
+    "Tải lên file (hỗ trợ: jpg, jpeg, png, mp4, mp3, wav, pdf):",
+    type=["jpg", "jpeg", "png", "mp4", "mp3", "wav", "pdf"],
+    accept_multiple_files=True
+)
+
+attachments = []
+
+if uploaded_files:
+    for file in uploaded_files:
+        file_name = file.name
+        file_ext = file_name.split('.')[-1].lower()
+        file_bytes = file.read()
+        
+        # Với file PDF, ta có thể trích xuất nội dung text
+        if file_ext == "pdf":
+            try:
+                import PyPDF2
+                pdf_reader = PyPDF2.PdfReader(io.BytesIO(file_bytes))
+                pdf_text = ""
+                for page in pdf_reader.pages:
+                    pdf_text += page.extract_text() or ""
+                attachments.append({
+                    "type": "pdf",
+                    "data": pdf_text
+                })
+            except Exception as e:
+                st.error(f"Lỗi khi đọc PDF {file_name}: {e}")
+        else:
+            # Các file khác: chuyển sang base64 (giả định API hỗ trợ dạng này)
+            encoded_file = base64.b64encode(file_bytes).decode('utf-8')
+            if file_ext in ["jpg", "jpeg", "png"]:
+                attachments.append({
+                    "type": "image",
+                    "data": encoded_file
+                })
+            elif file_ext == "mp4":
+                attachments.append({
+                    "type": "video",
+                    "data": encoded_file
+                })
+            elif file_ext in ["mp3", "wav"]:
+                attachments.append({
+                    "type": "audio",
+                    "data": encoded_file
+                })
+
+# Nút gửi yêu cầu
+if st.button("Gửi"):
+    if not user_prompt and not attachments:
+        st.warning("Vui lòng nhập nội dung chat hoặc upload file.")
+    else:
+        st.spinner("Đang xử lý yêu cầu ...")
+        try:
+            # Gọi API của Gemini với prompt và các attachments
+            response = genai.chat(
+                prompt=user_prompt,
+                attachments=attachments,  # Nếu không có file, gửi rỗng list
+                model=MODEL_ID
+            )
+            st.write("**Phản hồi từ Gemini:**")
+            st.write(response)
+        except Exception as ex:
+            st.error(f"Có lỗi xảy ra: {ex}")