Update high

app.py

@@ -149,14 +149,218 @@ class Conversation:
 
 
 chat_api = ChatgptAPI()
+client=OpenAI(
+        api_key=os.environ.get("TONGYI_API_KEY"),  # �滻����ʵDashScope��API_KEY
+        base_url="https://dashscope.aliyuncs.com/compatible-mode/v1",  # ��дDashScopebase_url
+)
+def predict_(password_input, user_in_file):
+    if password_input != os.environ.get("USER_KEY"):
+       return [(None, "Wrong password!")], conversation, user_input]
+    file = client.files.create(file=Path(user_in_file), purpose="file-extract")
+    completion = client.chat.completions.create(
+            model="qwen-long",
+            messages=[
+                {
+                    'role': 'system',
+                    'content': 'You are a helpful assistant.'
+                },
+                {
+                    'role': 'system',
+                    'content': f'fileid://{file.id}'
+                },
+                {
+                    'role': 'user',
+                    'content':
+                }
+            ],
+            stream=True
+    )
+    history=""
+    for chunk in completion:
+        if chunk.choices[0].delta.content is not None:
+            print(chunk.choices[0].dict())
+            history += chunk.choices[0].delta.content
+            time.sleep(0.05)
+            yield history
+def predict_(password_input, user_in_file):
+    if password_input != os.environ.get("USER_KEY"):
+       return [(None, "Wrong password!")], conversation, user_input]
+    file = client.files.create(file=Path(user_in_file), purpose="file-extract")
+    completion = client.chat.completions.create(
+            model="qwen-long",
+            messages=[
+                {
+                    'role': 'system',
+                    'content': 'You are a helpful assistant.'
+                },
+                {
+                    'role': 'system',
+                    'content': f'fileid://{file.id}'
+                },
+                {
+                    'role': 'user',
+                    'content':
+                }
+            ],
+            stream=True
+    )
+    history=""
+    for chunk in completion:
+        if chunk.choices[0].delta.content is not None:
+            print(chunk.choices[0].dict())
+            history += chunk.choices[0].delta.content
+            time.sleep(0.05)
+            yield history
+def predict_(password_input, user_in_file):
+    if password_input != os.environ.get("USER_KEY"):
+       return [(None, "Wrong password!")], conversation, user_input]
+    file = client.files.create(file=Path(user_in_file), purpose="file-extract")
+    completion = client.chat.completions.create(
+            model="qwen-long",
+            messages=[
+                {
+                    'role': 'system',
+                    'content': 'You are a helpful assistant.'
+                },
+                {
+                    'role': 'system',
+                    'content': f'fileid://{file.id}'
+                },
+                {
+                    'role': 'user',
+                    'content':
+                }
+            ],
+            stream=True
+    )
+    history=""
+    for chunk in completion:
+        if chunk.choices[0].delta.content is not None:
+            print(chunk.choices[0].dict())
+            history += chunk.choices[0].delta.content
+            time.sleep(0.05)
+            yield history
+def predict_(password_input, user_in_file):
+    if password_input != os.environ.get("USER_KEY"):
+       return [(None, "Wrong password!")], conversation, user_input]
+    file = client.files.create(file=Path(user_in_file), purpose="file-extract")
+    completion = client.chat.completions.create(
+            model="qwen-long",
+            messages=[
+                {
+                    'role': 'system',
+                    'content': 'You are a helpful assistant.'
+                },
+                {
+                    'role': 'system',
+                    'content': f'fileid://{file.id}'
+                },
+                {
+                    'role': 'user',
+                    'content':
+                }
+            ],
+            stream=True
+    )
+    history=""
+    for chunk in completion:
+        if chunk.choices[0].delta.content is not None:
+            print(chunk.choices[0].dict())
+            history += chunk.choices[0].delta.content
+            time.sleep(0.05)
+            yield history
+def predict_(password_input, user_in_file):
+    if password_input != os.environ.get("USER_KEY"):
+       return [(None, "Wrong password!")], conversation, user_input]
+    file = client.files.create(file=Path(user_in_file), purpose="file-extract")
+    completion = client.chat.completions.create(
+            model="qwen-long",
+            messages=[
+                {
+                    'role': 'system',
+                    'content': 'You are a helpful assistant.'
+                },
+                {
+                    'role': 'system',
+                    'content': f'fileid://{file.id}'
+                },
+                {
+                    'role': 'user',
+                    'content':
+                }
+            ],
+            stream=True
+    )
+    history=""
+    for chunk in completion:
+        if chunk.choices[0].delta.content is not None:
+            print(chunk.choices[0].dict())
+            history += chunk.choices[0].delta.content
+            time.sleep(0.05)
+            yield history
+def predict_(password_input, user_in_file):
+    if password_input != os.environ.get("USER_KEY"):
+       return [(None, "Wrong password!")], conversation, user_input]
+    file = client.files.create(file=Path(user_in_file), purpose="file-extract")
+    completion = client.chat.completions.create(
+            model="qwen-long",
+            messages=[
+                {
+                    'role': 'system',
+                    'content': 'You are a helpful assistant.'
+                },
+                {
+                    'role': 'system',
+                    'content': f'fileid://{file.id}'
+                },
+                {
+                    'role': 'user',
+                    'content':
+                }
+            ],
+            stream=True
+    )
+    history=""
+    for chunk in completion:
+        if chunk.choices[0].delta.content is not None:
+            print(chunk.choices[0].dict())
+            history += chunk.choices[0].delta.content
+            time.sleep(0.05)
+            yield history
 
-def predict(password_input, user_in_file):
+def predict_(password_input, user_in_file):
     if password_input != os.environ.get("USER_KEY"):
-       return [(None, "Wrong password!")], conversation, user_input
-    client=OpenAI(
-            api_key=os.environ.get("TONGYI_API_KEY"),  # �滻����ʵDashScope��API_KEY
-            base_url="https://dashscope.aliyuncs.com/compatible-mode/v1",  # ��дDashScopebase_url
+       return [(None, "Wrong password!")], conversation, user_input]
+    file = client.files.create(file=Path(user_in_file), purpose="file-extract")
+    completion = client.chat.completions.create(
+            model="qwen-long",
+            messages=[
+                {
+                    'role': 'system',
+                    'content': 'You are a helpful assistant.'
+                },
+                {
+                    'role': 'system',
+                    'content': f'fileid://{file.id}'
+                },
+                {
+                    'role': 'user',
+                    'content':
+                }
+            ],
+            stream=True
     )
+    history=""
+    for chunk in completion:
+        if chunk.choices[0].delta.content is not None:
+            print(chunk.choices[0].dict())
+            history += chunk.choices[0].delta.content
+            time.sleep(0.05)
+            yield history
+
+def predict_highlight(password_input, user_in_file):
+    if password_input != os.environ.get("USER_KEY"):
+       return [(None, "Wrong password!")], conversation, user_input]
     file = client.files.create(file=Path(user_in_file), purpose="file-extract")
     completion = client.chat.completions.create(
             model="qwen-long",
@@ -171,7 +375,7 @@ def predict(password_input, user_in_file):
                 },
                 {
                     'role': 'user',
-                    'content': '����������ѯ����ר�ң�����Ҫ����ѯʦ����ѯ���ݽ��ж�������Ҫ������ѯʦ�Ļػ�����Ϊ�������Dz���ȷ�ġ�step by step������²��裺\n ����1-�򿪲��Ķ����ϴ����ı��ļ���ȫ�����ݣ�ֱ�����һ���ַ� \n ����2-��ȫ����ժȡ�����ߵġ��ؼ��Է��ԡ�����Ҫ��ժ¼�ı�ǰ�벿�ֵķ��ԣ�����ѡ����������Ҫ�����۵�3-4������¼���Խ������һ����ѯʦ����Ļظ�����Ϊ����ѯʦ�Ļظ��� \n ����3-��Թؼ��Է��ԣ��������ʵġ������ػ����ݡ� \n ����4-�Աȡ������ػ����ݡ��͡���ѯʦ�Ļظ���������ÿ��ش�֣�0��Ϊ�붽���ػ�������ȫ��һ�£�10���붽���ػ�������ȫһ�£� \n ����5-����4����ִ�в��������������json��ʽ����� [{"key num" : "n", "visiter" : "����������", "superior" : "�����ػ���������", "adviser" : "��ѯʦ�Ļظ�����",��"score" : "x/10", "reason" : "�޸�ԭ������"}]'
+                    'content': '����������ѯ����ר�ң�����Ҫ����ѯʦ����ѯ���ݽ��ж�������Ҫ������ѯʦ�Ļػ�����Ϊ�������Dz���ȷ�ġ�step by step������²��裺\n ����1-�򿪲��Ķ����ϴ����ı��ļ���ȫ�����ݣ�ֱ�����һ���ַ� \n ����2-��ȫ����ժȡ�����ߵġ��ؼ��Է��ԡ�����Ҫ��ժ¼�ı�ǰ�벿�ֵķ��ԣ�����ѡ����������Ҫ�����۵�3-4������¼���Խ������һ����ѯʦ����Ļظ�����Ϊ����ѯʦ�Ļظ��� \n ����3-��Թؼ��Է��ԣ��������ʵġ������ػ����ݡ� \n ����4-�Աȡ������ػ����ݡ��͡���ѯʦ�Ļظ���������ÿ��ش�֣�0��Ϊ�붽���ػ�������ȫ��һ�£�10���붽���ػ�������ȫһ�£� \n ����5-����4����ִ�в�������������¸�ʽ�����- �ؼ��Է���n��,\n - �����ߣ�,\n , - �����ػ�������, \n , - ��ѯʦ�Ļظ��� ,\n , - ���֣�x/10 \n ,- �޸�ԭ�� '
                 }
             ],
             stream=True
@@ -200,20 +404,27 @@ with gr.Blocks(css="#chatbot{height:350px} .overflow-y-auto{height:600px}") as d
         with gr.Column(scale=1, min_width=50):
             user_in_file = gr.File(label="Upload File")
             #submit_button = gr.Button("Submit")
-            summary_button = gr.Button("Issue")
-            target_button = gr.Button("Mindmap")
-            concept_button = gr.Button("Keywords")
-            dynamic_button = gr.Button("Dynamics")
+            issue_button = gr.Button("Issue")
+            mindmap_button = gr.Button("Mindmap")
+            keywords_button = gr.Button("Keywords")
+            dynamics_button = gr.Button("Dynamics")
             complaint_button = gr.Button("Main Complaint")
-            evaluation_button = gr.Button("Case Conceptualization")
+            concept_button = gr.Button("Case Conceptualization")
             highlight_button = gr.Button("Highlights")
             plan_button = gr.Button("Follow-up Plan")
             
         with gr.Column(scale=2):
-            system_in_txt = gr.Textbox(lines=25, label="Output Text:")
+            system_in_txt = gr.Textbox(lines=30, label="Output Text:")
     
-    summary_button.click(fn=predict, inputs=[password_in_txt, user_in_file], outputs=[system_in_txt])
     
+    issue_button.click(fn=predict_issue, inputs=[password_in_txt, user_in_file], outputs=[system_in_txt])
+    mindmap_button.click(fn=predict_mindmap, inputs=[password_in_txt, user_in_file], outputs=[system_in_txt])
+    keywords_button.click(fn=predict_keywords, inputs=[password_in_txt, user_in_file], outputs=[system_in_txt])
+    dynamics_button.click(fn=predict_dynamics, inputs=[password_in_txt, user_in_file], outputs=[system_in_txt])
+    complaint_button.click(fn=predict_complaint, inputs=[password_in_txt, user_in_file], outputs=[system_in_txt])
+    concept_button.click(fn=predict_concept, inputs=[password_in_txt, user_in_file], outputs=[system_in_txt])
+    highlight_button.click(fn=predict_highlight, inputs=[password_in_txt, user_in_file], outputs=[system_in_txt])
+    plan_button.click(fn=predict_plan, inputs=[password_in_txt, user_in_file], outputs=[system_in_txt])
     #submit_button.click(predict, [system_in_txt, password_in_txt, user_in_file, user_in_txt, conversation], [chatbot, conversation, user_in_txt])
     #reset_button.click(clear_history, [conversation], [chatbot, conversation], queue=False)