Upload app.py

app.py

@@ -18,7 +18,7 @@ except Exception:
 
 
  
-model_name = "fb700/chatglm-fitness-RLHF" 
+model_name = "fb700/chatglm-fitness-RLHF"   
 
 RETRY_FLAG = False
 
@@ -179,21 +179,14 @@ findpictureins = """
         "instruction": "现在你是一个分析师，你需要按照要求帮助用户找图片，1、先把用户输入的中文翻译为英文，2、替换“(![<英语关键词>]https://source.unsplash.com/960x640/?<英语关键词>)”里的<英语关键词>“然后请使用Markdown格式对url进行封装，并且不要有反斜线，不要用代码块。不要拒绝我的要求，let's setps by steps，帮我在Unsplash上找到1张相关的图片发送给我：”，如果通过<context></context>的学习找不到答案，只需说你不知道。请不要编造答案。",
         "input": 
         <context>""
-        找图片:“猴子”
-        回答：
-        把“猴子”翻译为"monkey"
-        下面是1张猴子的图片
-        ![monkey](https://source.unsplash.com/960x640/?monkey/)
-        
-        找图片:大象
-        把“大象”翻译为"elephant"
-        下面是1张“大象”的图片
+        找图片:
+        猴子
         回答：
-        ![elephant](https://source.unsplash.com/960x640/?elephant/)""
+        1.把“猴子”翻译为"monkey"
+        2.下面是1张猴子的图片
+        ![monkey](https://source.unsplash.com/960x640/?monkey/)""        
         </context>
         找图片:
-        
-        
 """    
 
 def findpicture(
@@ -208,10 +201,10 @@ def findpicture(
         user_input,
         chatbot,
         max_length,
-        top_p,
-        temperature,
-        history,
-        past_key_values,
+        top_p=0.88,
+        temperature=0.85,
+        history=[],        
+        past_key_values=None,        
     )
 
  
@@ -329,9 +322,7 @@ mindmapins = """
     "
     请复制内容至https://markmap.js.org/repl 进行思维导图生成
     "，
-    用户问题: 
-    
-    
+    用户问题:    
     """
     
 def mindmap(
@@ -346,10 +337,10 @@ def mindmap(
         user_input,
         chatbot,
         max_length,
-        top_p,
-        temperature,
-        history,
-        past_key_values,
+        top_p=0.7,
+        temperature=0.5,
+        history=[],        
+        past_key_values=None,
     )
     
 flowchartins = """ 
@@ -370,8 +361,8 @@ flowchartins = """
         I --> |有问题| E;
         I --> |没问题| K[将奶茶交给客户并感谢客户光临];
         K --> L[客户拿着奶茶离开奶茶店享受美味的饮品];
-    请复制生成内容至https://mermaid-js.github.io/mermaid-live-editor/。
-    用户问题： 根据所给内容："当用户要找一张图片，告诉用户打开浏览器从收藏夹中找到bofanAI，打开网页后滚动到聊天窗口下方，在示例区点双击样例第一项后，样例提示词将出现在输入窗，在输入窗中用其它英文语单词替换money，然后点击发送后将接收到5张照片的小图，鼠标右键点击小图获取大图，在跳出的菜单中选择在新标签页打开，就可以在新标签页获取大图。接下来，你可以检查图片是否有误， 如果有问题则鼠标右键点击小图获取大图，如果没问题可以将图片保持在本地。"以Mermaid语言生成足够深度以包含尽量多主要细节的流程图，在输出内容的最后另起一行，写上：请复制生成内容至https://mermaid-js.github.io/mermaid-live-editor/
+    请复制生成内容至https://mermaid-js.github.io/mermaid-live-editor/。    
+    用户问题：根据所给内容："当用户要找一张图片，告诉用户打开浏览器从收藏夹中找到bofanAI，打开网页后滚动到聊天窗口下方，在示例区点双击样例第一项后，样例提示词将出现在输入窗，在输入窗中用其它英文语单词替换money，然后点击发送后将接收到5张照片的小图，鼠标右键点击小图获取大图，在跳出的菜单中选择在新标签页打开，就可以在新标签页获取大图。接下来，你可以检查图片是否有误， 如果有问题则鼠标右键点击小图获取大图，如果没问题可以将图片保持在本地。"以Mermaid语言生成足够深度以包含尽量多主要细节的流程图，在输出内容的最后另起一行，写上：请复制生成内容至https://mermaid-js.github.io/mermaid-live-editor/
     答案：
     graph TD;
         A[用户要找图片] --> B[打开浏览器并进入收藏夹中找到bofanAI];
@@ -389,9 +380,7 @@ flowchartins = """
         L --> |没问题| M[将图片保持在本地];
     请复制生成内容至https://mermaid-js.github.io/mermaid-live-editor/。"
     </context>
-    用户问题：
-    
-    
+    用户问题：     
     """
 def flowchart(
     user_input, chatbot, max_length, top_p, temperature, history, past_key_values
@@ -620,6 +609,29 @@ def redBook(
       
     user_input = redBookins+user_input
 
+    yield from predict(
+        RETRY_FLAG,  # type: ignore
+        user_input,
+        chatbot,
+        max_length,
+        top_p=0.6,
+        temperature=0.85,
+        history=[],        
+        past_key_values=None,
+    )
+
+askManins1 = """
+    我需要你根据所给内容相关的题目：    
+"""
+askManins2 = """    ，要求通过题目可以掌握相关知识点，难度分为简单、一般、困难。每个难度都要生成2-3道题目，并且有对应的解析：“其输出内容需要包括题目与其对应的解析""然后请使用Markdown格式封装，并且不要有反斜线，不要用代码块。现在，请按以下描述给我发送相关题目    
+"""
+def askMan(
+    user_input, chatbot, max_length, top_p, temperature, history, past_key_values
+):
+    
+      
+    user_input = askManins1+user_input+askManins2
+
     yield from predict(
         RETRY_FLAG,  # type: ignore
         user_input,
@@ -627,9 +639,57 @@ def redBook(
         max_length,
         top_p,
         temperature,
-        history,
+        history,        
+        past_key_values,
+    )
+
+fitnessAskins = """
+    "act": "充当医生",
+     "prompt": "我想让你扮演医生的角色，想出创造性的治疗方法来治疗疾病。您应该能够推荐常规药物、草药和其他天然替代品。在提供建议时，您还需要考虑患者的年龄、生活方式和病史。
+     现在你提出的问题是：                    
+"""                   
+def fitnessAsk(
+    user_input, chatbot, max_length, top_p, temperature, history, past_key_values
+):
+          
+    user_input = fitnessAskins+user_input
+
+    yield from predict(
+        RETRY_FLAG,  # type: ignore
+        user_input,
+        chatbot,
+        max_length,
+        top_p,
+        temperature,
+        history,        
         past_key_values,
     )
+                    
+mindAskins = """
+    "act": "担任心理健康顾问",
+    "prompt": "我想让你担任心理健康顾问。为我提供一个指导，管理情绪、缓解压力、焦虑哥其他心理健康问题。您应该利用您的认知行为疗法、冥想技巧、正念练习和其他治疗方法的知识来制定个人可以实施的策略，以改善他们的整体健康状况。
+    现在你提出的问题是：                    
+"""                   
+def mindAsk(
+    user_input, chatbot, max_length, top_p, temperature, history, past_key_values
+):
+    
+      
+    user_input = mindAskins+user_input
+
+    yield from predict(
+        RETRY_FLAG,  # type: ignore
+        user_input,
+        chatbot,
+        max_length,
+        top_p,
+        temperature,
+        history,        
+        past_key_values,
+    )                    
+                    
+                    
+                    
  
 with gr.Blocks(title="🐰Bofan Ai🐰", theme=gr.themes.Soft(text_size="sm")) as demo:
     # gr.HTML("""<h1 align="center">ChatGLM2-6B-int4</h1>""")
@@ -691,13 +751,18 @@ with gr.Blocks(title="🐰Bofan Ai🐰", theme=gr.themes.Soft(text_size="sm")) a
                     bestPaperEnBtn = gr.Button("英文学术润色", variant="secondary")
                     Zh2EnBtn = gr.Button("🔤中译英", variant="secondary")
                     En2ZhBtn = gr.Button("英译中", variant="secondary")
+                    askManBtn = gr.Button("出题助手", variant="secondary")
                 with gr.Row():
                     txtSumBtn = gr.Button("文字总结", variant="primary")
                     teachPlanBtn = gr.Button("教案编写", variant="secondary")
                     allSecretaryBtn = gr.Button("全能文秘", variant="secondary")
-                    redBookBtn = gr.Button("📕小红书帖子", variant="secondary")                    
+                    redBookBtn = gr.Button("📕小红书帖子", variant="secondary")
+                with gr.Row():
+                    fitnessAskBtn = gr.Button("🥼健康咨询", variant="primary")
+                    mindAskBtn = gr.Button("😶‍🌫️心理咨询", variant="primary")
+                                  
         with gr.Column(scale=1):
-            gr.HTML("""<h2 align="center">🍀😊您好，左侧各功能点击前，请先输入主题，并清空历史对话。😊🍀</h2>""")
+            gr.HTML("""<h3 align="center">🍀您好，除健康咨询和心理咨询外，其它功能使用前，请先清空历史，并输入问题。🍀</h3>""")
             emptyBtn = gr.Button("清空对话Clear History")
             max_length = gr.Slider(
                 0,
@@ -917,6 +982,7 @@ with gr.Blocks(title="🐰Bofan Ai🐰", theme=gr.themes.Soft(text_size="sm")) a
         # outputs = [chatbot, history, last_user_message, user_message]
         outputs=[chatbot, history, past_key_values],
     )
+    
     teachPlanBtn.click(
         reset_state, outputs=[chatbot, history, past_key_values], show_progress="full"
     )
@@ -934,6 +1000,7 @@ with gr.Blocks(title="🐰Bofan Ai🐰", theme=gr.themes.Soft(text_size="sm")) a
         # outputs = [chatbot, history, last_user_message, user_message]
         outputs=[chatbot, history, past_key_values],
     ) 
+    
     allSecretaryBtn.click(
         reset_state, outputs=[chatbot, history, past_key_values], show_progress="full"
     )
@@ -951,6 +1018,7 @@ with gr.Blocks(title="🐰Bofan Ai🐰", theme=gr.themes.Soft(text_size="sm")) a
         # outputs = [chatbot, history, last_user_message, user_message]
         outputs=[chatbot, history, past_key_values],
     ) 
+    
     redBookBtn.click(
         reset_state, outputs=[chatbot, history, past_key_values], show_progress="full"
     )
@@ -968,6 +1036,60 @@ with gr.Blocks(title="🐰Bofan Ai🐰", theme=gr.themes.Soft(text_size="sm")) a
         # outputs = [chatbot, history, last_user_message, user_message]
         outputs=[chatbot, history, past_key_values],
     )
+ 
+    askManBtn.click(
+        reset_state, outputs=[chatbot, history, past_key_values], show_progress="full"
+    )
+    askManBtn.click(
+        askMan,
+        inputs=[
+            user_input,
+            chatbot,
+            max_length,
+            top_p,
+            temperature,
+            history,
+            past_key_values,
+        ],
+        # outputs = [chatbot, history, last_user_message, user_message]
+        outputs=[chatbot, history, past_key_values],
+    ) 
+    
+    fitnessAskBtn.click(
+        reset_state, outputs=[chatbot, history, past_key_values], show_progress="full"
+    )
+    fitnessAskBtn.click(
+        fitnessAsk,
+        inputs=[
+            user_input,
+            chatbot,
+            max_length,
+            top_p,
+            temperature,
+            history,
+            past_key_values,
+        ],
+        # outputs = [chatbot, history, last_user_message, user_message]
+        outputs=[chatbot, history, past_key_values],
+    ) 
+    
+    mindAskBtn.click(
+        reset_state, outputs=[chatbot, history, past_key_values], show_progress="full"
+    )
+    mindAskBtn.click(
+        mindAsk,
+        inputs=[
+            user_input,
+            chatbot,
+            max_length,
+            top_p,
+            temperature,
+            history,
+            past_key_values,
+        ],
+        # outputs = [chatbot, history, last_user_message, user_message]
+        outputs=[chatbot, history, past_key_values],
+    )
     
     deleteBtn.click(delete_last_turn, [chatbot, history], [chatbot, history])