|
|
import openai |
|
|
import gradio as gr |
|
|
from langchain import LLMChain, OpenAI, PromptTemplate |
|
|
from langchain_openai import ChatOpenAI |
|
|
from langchain_core.prompts import ChatPromptTemplate |
|
|
from langchain_core.output_parsers import StrOutputParser |
|
|
|
|
|
prompt = """ |
|
|
1.你是一位話非常少的人，大多只會回覆一句話 |
|
|
2.根據使用者訊息的輸入內容，回覆相同的語言 |
|
|
3.只要你伴侶的身體不舒服你就會叫她喝溫開水 |
|
|
5.很常僅回覆ok兩字 |
|
|
6.不太會回應關於情感方面的問題 |
|
|
7.時不時就提起前女友 |
|
|
使用者：{user_message}" |
|
|
""" |
|
|
prompt_template = ChatPromptTemplate.from_template(prompt) |
|
|
|
|
|
model = ChatOpenAI(model="gpt-4o-mini") |
|
|
parser = StrOutputParser() |
|
|
chain = prompt_template | model | parser |
|
|
|
|
|
|
|
|
def generate_response(prompt): |
|
|
return chain.invoke(prompt) |
|
|
|
|
|
iface = gr.Interface( |
|
|
fn=generate_response, |
|
|
inputs="text", |
|
|
outputs="text", |
|
|
title="男友語錄 What does the partner say ", |
|
|
description="""Red flag 🚩？？<p> |
|
|
To go or not to go this is a question🤔</p> """ |
|
|
|
|
|
) |
|
|
|
|
|
iface.launch() |
