Update app.py

app.py

@@ -8,9 +8,15 @@ from langchain.tools import Tool,YoutubeSearchTool as YTS
 from langchain.memory import ConversationalBufferMomory as MEM,RedisChatHistory as HIS
 from langchain.schema import SystemMessage as SM,HumanMessage as HM
 from langchain import hub
-import chainlit as cl
 import os
 from langchain.retrievers import WikipediaRetriever as Wiki
+import gradio as gr
+chatbot = gr.Chatbot(
+    label="SYAI4.1",
+    show_copy_button=True,
+    likeable=True,
+    layout="panel"
+)
 def terminal(c):
     a=Popen(c,shell=True,stdin=P,stdout=P,stderr=P)
     return a.stdout.read()+a.stderr.read()
@@ -23,26 +29,42 @@ tools.append(Tool.from_function(func=terminal,name="terminal"))
 tools.append(crt(name="wiki",description="위키 백과를 검색하여 정보를 가져온다",retriever=Wiki(lang="ko",top_k_results=1)))
 llm=HF(repo_id="peterpeter8585/syai4.0")
 prompt=hub.pull("hwchase17/structed-chat-agent")
-@cl.on_chat_start
-async def on_chat_start():
-    thread_id=None
-    while not thread_id:
-        res=await cl.AskUserMessage(content="Please enter the username and the id.ex) peterpeter9595_chat1").send()
-        if res:
-            thread_id=res["content"]
-    history=HIS(session_id=thread_id, url=os.environ["URL"])
-    memory=MEM(chat_memory=history,memory_key="history")
-    m_r=memory.load_memory_variables({})["history"]
+def chat(message,
+    history: list[tuple[str, str]],
+    system_message,
+    max_tokens,
+    temperature,
+    top_p, chat_session=""):
+    messages=[SM(content=system_message+"And, Your name is Chatchat")]
+    for val in history:
+        if val[0]:
+            messages.append(HM(content=val[0]))
+        if val[1]:
+            messages.append(AM(content=val[1]))
+
+    messages.append(HM(content=message))
+    history1=HIS(session_id=chat_session, url=os.environ["URL"])
+    memory=MEM(chat_memory=history1,memory_key="history")
     agent=EX(agent=Agent(llm,tools,prompt,memory_key="history"),tools=tools,verbose=True,handle_parsing_errors=True)
+    yield agent.invoke(messages)
+ai1=gr.ChatInterface(
+    chat,
+    chatbot=chatbot,
+    additional_inputs=[
+        gr.Textbox(value="You are a helpful assistant.", label="System message",  interactive=True),
+        gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens"),
+        gr.Slider(minimum=0.1, maximum=4.0, value=0.1, step=0.1, label="Temperature"),
+        gr.Slider(
+            minimum=0.1,
+            maximum=1.0,
+            value=0.1,
+            step=0.05,
+            label="Top-p (nucleus sampling)",
+        ),
+        gr.Textbox(label="chat_id(please enter the chat id!)")
+    ],
     
-    for m in m_r:
-        if isinstance(m,HM):
-            await cl.Message(author="User",content=f"{m.content}").send()
-        else:
-            await cl.Message(author="ChatBot",content=f"{m.content}").send()
-    cl.user_session.set("agent",agent)
-@cl.on_message
-async def on_message(message:str):
-    exe=cl.user_session.get("agent")
-    result=exe(message)
-    await cl.Message(content=result["output"]).send()
+)
+with gr.Blocks(theme="shivi/calm_seafoam") as ai:
+    gr.TabbedInterface([ai1],["Chatchat"])
+    ai.launch()