Update app

app.py

@@ -1,81 +1,87 @@
 import streamlit as st
-import base64
 import io
-import google.genai as genai  # Import SDK của google-genai
+import PyPDF2
+from google import genai
+from google.genai import types
+import os
 
-# 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)
+# Sử dụng biến môi trường GOOGLE_API_KEY đã được thiết lập
+client = genai.Client()
+
+api_key = os.getenv('GOOGLE_API_KEY')
+if not api_key:
+    st.error("Biến môi trường GOOGLE_API_KEY chưa được thiết lập!")
+else:
+    st.write("API key đã được cấu hình.")
 
 st.title("Gemini 2.0 Pro Chatbot")
-st.write("Chào mừng đến với chatbot Gemini sử dụng google-genai SDK!")
+st.write("Chatbot sử dụng Google Gen AI SDK (Gemini Developer API)")
 
-# Nhập nội dung chat
-user_prompt = st.text_input("Nhập nội dung chat:")
+# Nhập nội dung chat từ người dùng
+user_text = st.text_input("Nhập nội dung chat:")
 
-# File uploader hỗ trợ nhiều loại file: hình ảnh, video, audio, PDF
+# Cho phép upload file với các định dạng: jpg, jpeg, png, mp4, mp3, wav, 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 = []
+# Tạo danh sách contents để gửi lên model
+contents = []
+
+if user_text:
+    contents.append(user_text)
 
 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
+        file_ext = file.name.split('.')[-1].lower()
         if file_ext == "pdf":
+            # Trích xuất nội dung văn bản từ 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
-                })
+                    page_text = page.extract_text()
+                    if page_text:
+                        pdf_text += page_text + "\n"
+                if pdf_text:
+                    contents.append(pdf_text)
+                else:
+                    st.warning(f"Không thể trích xuất nội dung từ {file.name}.")
             except Exception as e:
-                st.error(f"Lỗi khi đọc PDF {file_name}: {e}")
+                st.error(f"Lỗi khi xử lý 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
-                })
+            # Xác định mime type cho các file khác
+            if file_ext in ["jpg", "jpeg"]:
+                mime_type = "image/jpeg"
+            elif file_ext == "png":
+                mime_type = "image/png"
             elif file_ext == "mp4":
-                attachments.append({
-                    "type": "video",
-                    "data": encoded_file
-                })
-            elif file_ext in ["mp3", "wav"]:
-                attachments.append({
-                    "type": "audio",
-                    "data": encoded_file
-                })
+                mime_type = "video/mp4"
+            elif file_ext == "mp3":
+                mime_type = "audio/mpeg"
+            elif file_ext == "wav":
+                mime_type = "audio/wav"
+            else:
+                mime_type = "application/octet-stream"
+            # Tạo Part từ bytes
+            part = types.Part.from_bytes(data=file_bytes, mime_type=mime_type)
+            contents.append(part)
 
-# Nút gửi yêu cầu
 if st.button("Gửi"):
-    if not user_prompt and not attachments:
+    if not contents:
         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}")
+        with st.spinner("Đang xử lý yêu cầu..."):
+            try:
+                # Gọi API generate_content theo cách mẫu trong tài liệu
+                response = client.models.generate_content(
+                    model='gemini-2.0-pro-exp-02-05',
+                    contents=contents
+                )
+                st.write("**Phản hồi từ Gemini:**")
+                st.write(response.text)
+            except Exception as e:
+                st.error(f"Có lỗi xảy ra: {e}")