Upload 3 files

README.md

@@ -1,13 +1,13 @@
 ---
-title: Assistant Chatbot
-emoji: 👁
-colorFrom: pink
-colorTo: purple
+title: Chatbot
+emoji: 📉
+colorFrom: green
+colorTo: yellow
 sdk: streamlit
-sdk_version: 1.35.0
+sdk_version: 1.33.0
 app_file: app.py
 pinned: false
-license: unknown
+license: other
 ---
 
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -0,0 +1,74 @@
+import os
+import streamlit as st
+from langchain.chat_models import ChatOpenAI
+from langchain.schema import HumanMessage
+from langchain.agents import AgentType, initialize_agent, load_tools
+from langchain.callbacks import StreamlitCallbackHandler
+from langchain.memory import ConversationBufferMemory
+from langchain.prompts import MessagesPlaceholder
+
+st.title("文章生成支援Chatbot")
+
+api_model = os.getenv("OPENAI_API_MODEL")
+temperature = os.getenv("OPENAI_API_TEMPERATURE")
+
+origin_text = st.sidebar.text_area("プロンプト入力")
+system_prompt = origin_text if origin_text else os.getenv("system_prompt")
+print(system_prompt)
+
+def create_agent_chain():
+    chat = ChatOpenAI(
+            model_name = api_model,
+            temperature = temperature,
+            streaming = True,
+    )
+    
+    agent_kwargs = {
+        "extra_prompt_messages": [MessagesPlaceholder(variable_name = "memory")],
+    }
+    
+    memory = ConversationBufferMemory(memory_key = "memory", return_messages = True)
+    
+    tools = load_tools(["ddg-search", "wikipedia"])
+    return initialize_agent(
+        tools, 
+        chat, 
+        agent = AgentType.OPENAI_FUNCTIONS,
+        agent_kwargs = agent_kwargs, 
+        memory = memory, 
+    )
+
+if "messages" not in st.session_state:
+    st.session_state.messages = []
+
+if "agent_chain" not in st.session_state:
+    st.session_state.agent_chain = create_agent_chain()
+
+if "polly_client" not in st.session_state:
+    st.session_state.polly_client = PollyClient() 
+
+for message in st.session_state.messages:
+    with st.chat_message(message["role"]):
+        st.markdown(message["content"])
+
+prompt = st.chat_input("What is up?")
+
+if prompt:
+    st.session_state.messages.append({"role": "user", "content": prompt})
+    
+    with st.chat_message("user"):
+        st.markdown(prompt)
+    
+    with st.chat_message("assistant"):
+        callback = StreamlitCallbackHandler(st.container())
+        response = st.session_state.agent_chain.run(system_prompt + prompt, callbacks = [callback])
+        st.markdown(response)
+        
+        # 音声ファイルを生成
+        file_name = "response.mp3"
+        st.session_state.polly_client.text_to_speech(response, file_name)
+
+        audio_file_path = file_name
+        st.audio(audio_file_path)
+        
+    st.session_state.messages.append({"role": "assistant", "content": response})

requirements.txt

@@ -0,0 +1,7 @@
+boto3
+streamlit
+python-dotenv
+langchain 
+openai
+wikipedia==1.4.0
+duckduckgo-search==4.1.0