tawondazz/Mobile-Agent
0
1o
2 zf� � @ sJ d dl Z d dlmZ dd� Zdd� Zdd� Zdd d3�Zd d � Zd
d� ZdS )� N)� encode_imagec C �" g } d}| � dd|d�gg� | S )Nz�You are a helpful AI mobile phone operating assistant. You need to help me operate the phone to complete the user's instruction.�system�text��typer ��append�Zoperation_historyZ
sysetm_prompt� r �9E:\Desktop\Mobile-Agent-v2-demo-local\MobileAgent\chat.py�init_action_chat � r
c C r �Nz6You are a helpful AI mobile phone operating assistant.r r r r r4 r r r �init_reflect_chat r r c C r r r r5 r r r �init_memory_chat r r c C sT t �|�}|rt|�}d|d�ddd|� �id�g}nd|d�g}|�| |g� |S )Nr r � image_url�url�data:image/jpeg;base64,�r r ��copy�deepcopyr r )�role�prompt� chat_history�image�new_chat_historyZ base64_image�contentr r r � add_response s" 6�7�����r c C sd t �|�}t|d �}t|d �}d|d�ddd|� �id�ddd|� �id�g}|�| |g� |S ) Nr � r r r r r r r )r r r r r Z
base64_image1Z
base64_image2r r r r �add_response_two_image5 s" 8 �9��10���r! c C sX t d� | D ] }t d|d � t |d d d dt|d �d d � qt d� d S )Nzd****************************************************************************************************zrole:r r r z<image>�11)�print�len)r �chatr r r � print_statusQ s12 . r& )N) r �MobileAgent.apir r
r r r r! r& r r r r �<module> s 13