Update app.py

app.py

@@ -1,38 +1,26 @@
 import os
-import gradio as gr
-from langchain.chat_models import ChatOpenAI
-from langchain import LLMChain, PromptTemplate
-from langchain.memory import ConversationBufferMemory
+from flask import Flask, request, jsonify, send_file
+from flask_cors import CORS
 import openai
 from docx import Document
 from fpdf import FPDF
+import smtplib
+from email.mime.text import MIMEText
+from email.mime.multipart import MIMEMultipart
 
-# 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.
-{chat_history}
-User: {user_message}
-Chatbot:"""
-
-prompt = PromptTemplate(
-    input_variables=["chat_history", "user_message"], template=template
-)
-
-memory = ConversationBufferMemory(memory_key="chat_history")
+app = Flask(__name__)
+CORS(app)
 
-llm_chain = LLMChain(
-    llm=ChatOpenAI(temperature=0.5, model_name="gpt-4"),
-    prompt=prompt,
-    verbose=True,
-    memory=memory,
-)
-
-def get_text_response(user_message, chat_history):
-    response = llm_chain.predict(user_message=user_message)
-    return response
-
-def translate_text(text):
+# Lấy API key từ biến môi trường
+openai.api_key = os.getenv('OPENAI_API_KEY')
+
+@app.route('/translate', methods=['POST'])
+def translate():
+    data = request.json
+    text = data.get('text')
+    target_language = 'vi'  # Dịch sang tiếng Việt
+    
+    # Sử dụng OpenAI API để dịch
     prompt = f"Translate the following text to Vietnamese: {text}"
     response = openai.Completion.create(
         engine="gpt-4",
@@ -40,15 +28,22 @@ def translate_text(text):
         max_tokens=150
     )
     translation = response.choices[0].text.strip()
-    return translation
+    return jsonify({"translation": translation})
+
+@app.route('/create_file', methods=['POST'])
+def create_file():
+    data = request.json
+    text = data.get('text')
+    filename = data.get('filename')
+    file_format = data.get('file_format')
 
-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
+        return send_file(doc_path, as_attachment=True)
+
     elif file_format == 'pdf':
         pdf = FPDF()
         pdf.add_page()
@@ -56,46 +51,35 @@ def create_text_file(text, filename, file_format):
         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
+        return send_file(pdf_path, as_attachment=True)
+
+@app.route('/send_email', methods=['POST'])
+def send_email():
+    data = request.json
+    subject = data.get('subject')
+    body = data.get('body')
+    to_email = data.get('to_email')
+
+    # Thông tin đăng nhập email
+    from_email = os.getenv('EMAIL_USER')
+    email_password = os.getenv('EMAIL_PASS')
+
+    msg = MIMEMultipart()
+    msg['From'] = from_email
+    msg['To'] = to_email
+    msg['Subject'] = subject
+    msg.attach(MIMEText(body, 'plain'))
+
+    try:
+        server = smtplib.SMTP('smtp.gmail.com', 587)
+        server.starttls()
+        server.login(from_email, email_password)
+        text = msg.as_string()
+        server.sendmail(from_email, to_email, text)
+        server.quit()
+        return jsonify({"message": "Email sent successfully"})
+    except Exception as e:
+        return jsonify({"message": str(e)}), 500
 
 if __name__ == "__main__":
-    demo = chatbot_interface()
-    demo.launch()
+    app.run(host='0.0.0.0', port=5000)