Update app.py

app.py

@@ -3,7 +3,11 @@ import gradio as gr
 from langchain.chat_models import ChatOpenAI
 from langchain import LLMChain, PromptTemplate
 from langchain.memory import ConversationBufferMemory
+import openai
+from docx import Document
+from fpdf import FPDF
 
+# Lấy API key từ biến môi trường
 OPENAI_API_KEY = os.getenv('OPENAI_API_KEY')
 
 template = """You are a very helpful assistant in providing users with information and knowledge about statistics. Additionally, you are also an expert in translating specialized documents on statistics, artificial intelligence, and technology from foreign languages into Vietnamese in a complete, coherent manner while ensuring the original knowledge from the foreign documents is accurately conveyed.
@@ -18,17 +22,80 @@ prompt = PromptTemplate(
 memory = ConversationBufferMemory(memory_key="chat_history")
 
 llm_chain = LLMChain(
-    llm=ChatOpenAI(temperature=0.5, model_name="gpt-3.5-turbo"),
+    llm=ChatOpenAI(temperature=0.5, model_name="gpt-4"),
     prompt=prompt,
     verbose=True,
     memory=memory,
 )
 
-def get_text_response(user_message, history):
+def get_text_response(user_message, chat_history):
     response = llm_chain.predict(user_message=user_message)
     return response
 
-demo = gr.ChatInterface(get_text_response)
+def translate_text(text):
+    prompt = f"Translate the following text to Vietnamese: {text}"
+    response = openai.Completion.create(
+        engine="gpt-4o-mini",
+        prompt=prompt,
+        max_tokens=150
+    )
+    translation = response.choices[0].text.strip()
+    return translation
+
+def create_text_file(text, filename, file_format):
+    if file_format == 'docx':
+        doc = Document()
+        doc.add_paragraph(text)
+        doc_path = f"{filename}.docx"
+        doc.save(doc_path)
+        return doc_path
+    elif file_format == 'pdf':
+        pdf = FPDF()
+        pdf.add_page()
+        pdf.set_font("Arial", size=12)
+        pdf.multi_cell(0, 10, text)
+        pdf_path = f"{filename}.pdf"
+        pdf.output(pdf_path)
+        return pdf_path
+
+def chatbot_interface():
+    with gr.Blocks() as demo:
+        chatbot = gr.Chatbot()
+        msg = gr.Textbox(placeholder="Nhập tin nhắn...")
+        clear = gr.Button("Xóa hội thoại")
+        translate_input = gr.Textbox(placeholder="Nhập văn bản để dịch...")
+        translate_button = gr.Button("Dịch sang tiếng Việt")
+        translate_output = gr.Textbox(label="Bản dịch")
+        file_text = gr.Textbox(placeholder="Nhập văn bản để tạo file...")
+        file_name = gr.Textbox(placeholder="Nhập tên file (không bao gồm đuôi)...")
+        file_format = gr.Dropdown(choices=["docx", "pdf"], label="Chọn định dạng file")
+        create_file_button = gr.Button("Tạo File")
+        download = gr.File(label="Tải về file", interactive=False)
+
+        def respond(message, chat_history):
+            response = get_text_response(message, chat_history)
+            chat_history.append(("Người dùng", message))
+            chat_history.append(("Chatbot", response))
+            return chat_history, chat_history
+
+        def clear_chat():
+            return [], []
+
+        def translate_and_display(text):
+            translation = translate_text(text)
+            return translation
+
+        def create_and_return_file(text, name, format):
+            file_path = create_text_file(text, name, format)
+            return file_path
+
+        msg.submit(respond, [msg, chatbot], [chatbot, chatbot])
+        clear.click(clear_chat, [], [chatbot])
+        translate_button.click(translate_and_display, [translate_input], [translate_output])
+        create_file_button.click(create_and_return_file, [file_text, file_name, file_format], [download])
+
+    return demo
 
 if __name__ == "__main__":
-    demo.launch() # To create a public link, set `share=True` in `launch()`. To enable errors and logs
+    demo = chatbot_interface()
+    demo.launch()