import os os.environ["OTEL_TRACES_EXPORTER"] = "none" import gradio as gr import uuid from utils.chat import ChatLaborLaw # Function to initialize a new session and create chatbot instance for that session async def initialize_session(): session_id = str(uuid.uuid4())[:8] chatbot = ChatLaborLaw() # chatbot = Chat("gemini-2.0-flash") history = [] return "", session_id, chatbot, history # Function to handle user input and chatbot response async def chat_function(prompt, history, session_id, chatbot): if chatbot is None: return history, "", session_id, chatbot # Skip if chatbot not ready # Append the user's input to the message history history.append({"role": "user", "content": prompt}) # Get the response from the chatbot response = await chatbot.chat(prompt) # ใช้ await ได้แล้ว # Append the assistant's response to the message history history.append({"role": "assistant", "content": response}) return history, "", session_id, chatbot # Function to save feedback with chat history async def send_feedback(feedback, history, session_id, chatbot): os.makedirs("app/feedback", exist_ok=True) filename = f"app/feedback/feedback_{session_id}.txt" with open(filename, "a", encoding="utf-8") as f: f.write("=== Feedback Received ===\n") f.write(f"Session ID: {session_id}\n") f.write(f"Feedback: {feedback}\n") f.write("Chat History:\n") for msg in history: f.write(f"{msg['role']}: {msg['content']}\n") f.write("\n--------------------------\n\n") return "" # Clear feedback input # Create the Gradio interface with gr.Blocks(theme=gr.themes.Soft(primary_hue="amber")) as demo: gr.Markdown("# สอบถามเรื่องกฎหมายแรงงาน") # Initialize State session_state = gr.State() chatbot_instance = gr.State() chatbot_history = gr.State([]) # Chat UI chatbot_interface = gr.Chatbot(type="messages", label="Chat History") user_input = gr.Textbox(placeholder="Type your message here...", elem_id="user_input", lines=1) submit_button = gr.Button("Send") clear_button = gr.Button("Delete Chat History") # Submit actions submit_button.click( fn=chat_function, inputs=[user_input, chatbot_history, session_state, chatbot_instance], outputs=[chatbot_interface, user_input, session_state, chatbot_instance] ) user_input.submit( fn=chat_function, inputs=[user_input, chatbot_history, session_state, chatbot_instance], outputs=[chatbot_interface, user_input, session_state, chatbot_instance] ) # # Clear history # clear_button.click(lambda: [], outputs=chatbot_interface) clear_button.click( fn=initialize_session, inputs=[], outputs=[user_input, session_state, chatbot_instance, chatbot_history] ).then( fn=lambda: gr.update(value=[]), inputs=[], outputs=chatbot_interface ) # Feedback section with gr.Row(): feedback_input = gr.Textbox(placeholder="Send us feedback...", label="Feedback") send_feedback_button = gr.Button("Send Feedback") send_feedback_button.click( fn=send_feedback, inputs=[feedback_input, chatbot_history, session_state, chatbot_instance], outputs=[feedback_input] ) # Initialize session on load demo.load( fn=initialize_session, inputs=[], outputs=[user_input, session_state, chatbot_instance, chatbot_history] ) # Launch demo.launch(share=True)