fix errors value

app.py

@@ -54,6 +54,7 @@ def initialize_dataset():
     """
     try:
         dataset = load_dataset(HF_DATASET, use_auth_token=HF_TOKEN)
+        print("Dataset đã tồn tại trên Huggingface Hub.")
     except Exception as e:
         print(f"Dataset chưa tồn tại. Tạo mới Dataset: {e}")
         # Tạo Dataset mới nếu chưa tồn tại
@@ -63,8 +64,12 @@ def initialize_dataset():
                 "messages": []
             })
         })
-        dataset.push_to_hub(HF_DATASET, private=True, token=HF_TOKEN)
-    return dataset
+        try:
+            dataset.push_to_hub(HF_DATASET, private=True, token=HF_TOKEN)
+            print("Dataset mới đã được tạo và đẩy lên Huggingface Hub.")
+        except Exception as push_e:
+            print(f"Lỗi khi đẩy Dataset lên Hub: {push_e}")
+    return
 
 def save_conversation(user_id: str, messages: List[Tuple[str, str]]):
     """
@@ -89,6 +94,7 @@ def save_conversation(user_id: str, messages: List[Tuple[str, str]]):
         updated_dataset = concatenate_datasets([dataset, new_dataset])
         # Đẩy lên Hub
         updated_dataset.push_to_hub(HF_DATASET, split="conversations", token=HF_TOKEN)
+        print(f"Cuộc hội thoại của user_id {user_id} đã được lưu.")
     except Exception as e:
         print(f"Lỗi khi đẩy Dataset lên Hub: {e}")
 
@@ -451,54 +457,6 @@ EXAMPLES = [
 ]
 
 # Cấu hình giao diện trò chuyện của Gradio với giao diện đẹp mắt
-chat_interface = gr.ChatInterface(
-    fn=generate,  # Hàm được gọi khi có tương tác từ người dùng
-    additional_inputs=[
-        gr.Slider(
-            label="Số token mới tối đa",
-            minimum=1,
-            maximum=MAX_MAX_NEW_TOKENS,
-            step=1,
-            value=DEFAULT_MAX_NEW_TOKENS,
-        ),
-        gr.Slider(
-            label="Nhiệt độ",
-            minimum=0.1,
-            maximum=4.0,
-            step=0.1,
-            value=0.6,
-        ),
-        gr.Slider(
-            label="Top-p (nucleus sampling)",
-            minimum=0.05,
-            maximum=1.0,
-            step=0.05,
-            value=0.9,
-        ),
-        gr.Slider(
-            label="Top-k",
-            minimum=1,
-            maximum=1000,
-            step=1,
-            value=50,
-        ),
-        gr.Slider(
-            label="Hình phạt sự lặp lại",
-            minimum=1.0,
-            maximum=2.0,
-            step=0.05,
-            value=1.2,
-        ),
-    ],
-    stop_btn=None,  # Không có nút dừng
-    examples=EXAMPLES,  # Các ví dụ được hiển thị cho người dùng
-    cache_examples=False,  # Không lưu bộ nhớ cache cho các ví dụ
-    title="🤖 OpenGPT-4o Chatbot",
-    description="Một trợ lý AI mạnh mẽ sử dụng mô hình Llama-3.2 cục bộ với các chức năng tìm kiếm web, tóm tắt văn bản và phân tích tâm lý.",
-    theme="default",  # Có thể thay đổi theme để giao diện đẹp hơn
-)
-
-# Tạo giao diện chính của Gradio với CSS tùy chỉnh
 with gr.Blocks(css="""
     .gradio-container {
         background-color: #f0f2f5;  /* Màu nền nhẹ nhàng */
@@ -521,10 +479,86 @@ with gr.Blocks(css="""
     }
 """, fill_height=True) as demo:
     gr.Markdown(DESCRIPTION)  # Hiển thị mô tả
+    
     # Define the state for user_id
     user_state = gr.State(None)
-    gr.DuplicateButton(value="Nhân bản Không gian để sử dụng riêng tư", elem_id="duplicate-button")  # Nút nhân bản không gian
-    chat_interface.render()
+    
+    # Nút nhân bản không gian
+    gr.DuplicateButton(value="Nhân bản Không gian để sử dụng riêng tư", elem_id="duplicate-button")
+    
+    # Chat Interface
+    with gr.Row():
+        chatbot = gr.Chatbot()
+    
+    with gr.Row():
+        with gr.Column():
+            message = gr.Textbox(
+                label="Bạn:",
+                placeholder="Nhập tin nhắn của bạn tại đây...",
+            )
+            submit = gr.Button("Gửi")
+        with gr.Column():
+            # Các thanh trượt cho tham số
+            max_new_tokens = gr.Slider(
+                label="Số token mới tối đa",
+                minimum=1,
+                maximum=MAX_MAX_NEW_TOKENS,
+                step=1,
+                value=DEFAULT_MAX_NEW_TOKENS,
+            )
+            temperature = gr.Slider(
+                label="Nhiệt độ",
+                minimum=0.1,
+                maximum=4.0,
+                step=0.1,
+                value=0.6,
+            )
+            top_p = gr.Slider(
+                label="Top-p (nucleus sampling)",
+                minimum=0.05,
+                maximum=1.0,
+                step=0.05,
+                value=0.9,
+            )
+            top_k = gr.Slider(
+                label="Top-k",
+                minimum=1,
+                maximum=1000,
+                step=1,
+                value=50,
+            )
+            repetition_penalty = gr.Slider(
+                label="Hình phạt sự lặp lại",
+                minimum=1.0,
+                maximum=2.0,
+                step=0.05,
+                value=1.2,
+            )
+    
+    # Kết nối nút gửi với hàm generate
+    submit.click(
+        generate,
+        inputs=[
+            message,
+            chatbot,
+            user_state,
+            max_new_tokens,
+            temperature,
+            top_p,
+            top_k,
+            repetition_penalty,
+        ],
+        outputs=chatbot,
+    )
+    
+    # Thêm các ví dụ
+    gr.Examples(
+        examples=EXAMPLES,
+        inputs=[message],
+        outputs=[chatbot],
+        fn=lambda x: x,  # Function to populate the message box with the example
+    )
 
+# Khởi chạy ứng dụng Gradio
 if __name__ == "__main__":
-    demo.queue(max_size=20).launch()  # Khởi chạy ứng dụng Gradio với hàng đợi kích thước tối đa là 20
+    demo.queue(max_size=20).launch(share=True)  # Đặt share=True để tạo liên kết công khai