4-19 1146

app.py

@@ -1,28 +1,57 @@
+
 from openai import OpenAI
-import gradio as gr 
+import gradio as gr
 import os
 
 API_KEY = os.getenv("PPLX_API_KEY")
 client = OpenAI(api_key=API_KEY, base_url="https://api.perplexity.ai")
 
-
 def predict(message, history):
     history_openai_format = []
     for human, assistant in history:
-        history_openai_format.append({"role": "user", "content": human })
-        history_openai_format.append({"role": "assistant", "content":assistant})
+        history_openai_format.append({"role": "user", "content": human})
+        history_openai_format.append({"role": "assistant", "content": assistant})
     history_openai_format.append({"role": "user", "content": message})
   
-    # response = client.chat.completions.create(model='mistral-7b-instruct',
-    response = client.chat.completions.create(model='pplx-70b-online',
-    messages= history_openai_format,
-    temperature=1.0,
-    stream=True)
+    # 使用 Perplexity AI 的 70B 模型进行响应
+    response = client.chat.completions.create(
+        model='pplx-70b-online',
+        messages=history_openai_format,
+        temperature=1.0,
+        stream=True
+    )
 
     partial_message = ""
     for chunk in response:
         if chunk.choices[0].delta.content is not None:
-              partial_message = partial_message + chunk.choices[0].delta.content
-              yield partial_message
+            partial_message += chunk.choices[0].delta.content
+            yield partial_message
+
+def update_text(show_text1, show_text2):
+    text1 = "text 1" if show_text1 else ""
+    text2 = "text 2" if show_text2 else ""
+    return f"{text1}\n{text2}"
+
+def change_textbox(choice):
+    #根据不同输入对输出控件进行更新
+    if choice == "short":
+        return gr.update(lines=2, visible=True, value="Short story: ")
+    elif choice == "long":
+        return gr.update(lines=8, visible=True, value="Long story...")
+    else:
+        return gr.update(visible=False)
+
+with gr.Blocks(fill_height=True) as demo:
+        radio = gr.Radio(
+            ["short", "long", "none"], label="Essay Length to Write?"
+        )
+        text = gr.Textbox(lines=2, interactive=True)
+        radio.change(fn=change_textbox, inputs=radio, outputs=text)
+        chat_interface = gr.ChatInterface(predict)
+
+
+# chat_interface = gr.ChatInterface(predict)
+# 启动应用
+demo.launch(share=True)
+
 
-gr.ChatInterface(predict).launch(share=True)