瀏覽代碼

Undo local linter style fixes

Cam 1 年之前
父節點
當前提交
c52379e76b
共有 1 個文件被更改,包括 30 次插入46 次删除
  1. 30 46
      examples/chat_with_ai/main.py

+ 30 - 46
examples/chat_with_ai/main.py

@@ -7,11 +7,9 @@ from langchain.chat_models import ChatOpenAI
 from nicegui import Client, ui
 from log_callback_handler import NiceGuiLogElementCallbackHandler
 
-OPENAI_API_KEY = "not-set"  # TODO: set your OpenAI API key here
+OPENAI_API_KEY = 'not-set'  # TODO: set your OpenAI API key here
 
-llm = ConversationChain(
-    llm=ChatOpenAI(model_name="gpt-3.5-turbo", openai_api_key=OPENAI_API_KEY),
-)
+llm = ConversationChain(llm=ChatOpenAI(model_name='gpt-3.5-turbo', openai_api_key=OPENAI_API_KEY))
 
 messages: List[Tuple[str, str]] = []
 thinking: bool = False
@@ -20,65 +18,51 @@ thinking: bool = False
 @ui.refreshable
 async def chat_messages() -> None:
     for name, text in messages:
-        ui.chat_message(text=text, name=name, sent=name == "You")
+        ui.chat_message(text=text, name=name, sent=name == 'You')
     if thinking:
-        ui.spinner(size="3rem").classes("self-center")
-    await ui.run_javascript(
-        "window.scrollTo(0, document.body.scrollHeight)", respond=False
-    )
+        ui.spinner(size='3rem').classes('self-center')
+    await ui.run_javascript('window.scrollTo(0, document.body.scrollHeight)', respond=False)
 
 
-@ui.page("/")
+@ui.page('/')
 async def main(client: Client):
-    with ui.tabs().classes("w-full") as tabs:
-        chat_panel = ui.tab("Chat")
-        logs_panel = ui.tab("Logs")
-    with ui.tab_panels(tabs).classes("w-full"):
+
+    with ui.tabs().classes('w-full') as tabs:
+        chat_panel = ui.tab('Chat')
+        logs_panel = ui.tab('Logs')
+    with ui.tab_panels(tabs).classes('w-full'):
         with ui.tab_panel(logs_panel):
-            log = ui.log().classes("w-full h-full")
+            log = ui.log().classes('w-full').style('height: 50vh;')
 
     async def send() -> None:
         global thinking
         message = text.value
-        messages.append(("You", text.value))
+        messages.append(('You', text.value))
         thinking = True
-        text.value = ""
+        text.value = ''
         chat_messages.refresh()
 
-        response = await llm.arun(
-            message, callbacks=[NiceGuiLogElementCallbackHandler(log)]
-        )
-        messages.append(("Bot", response))
+        response = await llm.arun(message, callbacks=[NiceGuiLogElementCallbackHandler(log)])
+        messages.append(('Bot', response))
         thinking = False
         chat_messages.refresh()
 
-    anchor_style = r"a:link, a:visited {color: inherit !important; text-decoration: none; font-weight: 500}"
-    ui.add_head_html(f"<style>{anchor_style}</style>")
+    anchor_style = r'a:link, a:visited {color: inherit !important; text-decoration: none; font-weight: 500}'
+    ui.add_head_html(f'<style>{anchor_style}</style>')
     await client.connected()
-
-    with ui.tab_panels(tabs, value=chat_panel).classes("w-full"):
+    
+    with ui.tab_panels(tabs, value=chat_panel).classes('w-full'):
         with ui.tab_panel(chat_panel):
-            with ui.column().classes("w-full max-w-2xl mx-auto items-stretch"):
+            with ui.column().classes('w-full max-w-2xl mx-auto items-stretch'):
                 await chat_messages()
 
-    with ui.footer().classes("bg-white"), ui.column().classes(
-        "w-full max-w-3xl mx-auto my-6"
-    ):
-        with ui.row().classes("w-full no-wrap items-center"):
-            placeholder = (
-                "message"
-                if OPENAI_API_KEY != "not-set"
-                else "Please provide your OPENAI key in the Python script first!"
-            )
-            text = (
-                ui.input(placeholder=placeholder)
-                .props("rounded outlined input-class=mx-3")
-                .classes("w-full self-center")
-                .on("keydown.enter", send)
-            )
-        ui.markdown("simple chat app built with [NiceGUI](https://nicegui.io)").classes(
-            "text-xs self-end mr-8 m-[-1em] text-primary"
-        )
-
+    with ui.footer().classes('bg-white'), ui.column().classes('w-full max-w-3xl mx-auto my-6'):
+        with ui.row().classes('w-full no-wrap items-center'):
+            placeholder = 'message' if OPENAI_API_KEY != 'not-set' else \
+                'Please provide your OPENAI key in the Python script first!'
+            text = ui.input(placeholder=placeholder).props('rounded outlined input-class=mx-3') \
+                .classes('w-full self-center').on('keydown.enter', send)
+        ui.markdown('simple chat app built with [NiceGUI](https://nicegui.io)') \
+            .classes('text-xs self-end mr-8 m-[-1em] text-primary')
 
-ui.run(title="Chat with GPT-3 (example)")
+ui.run(title='Chat with GPT-3 (example)')