Upload app.py

app.py

@@ -3,22 +3,52 @@ from openai import OpenAI
 import json
 import os
 
+api_key = os.environ.get("api_key")
+base_url = os.environ.get("mistrial_api")
+client = OpenAI(
+    api_key=api_key,
+    base_url=base_url
+)
+
 def test_api():
-    api_key = os.environ.get("api_key")
-    base_url = os.environ.get("mistrial_api")
-    client = OpenAI(
-        api_key=api_key,
-        base_url=base_url
+    models = client.models.list()
+    return str(models.model_dump())
+
+
+def generate(user_prompt,system_prompt, temperature, top_p):
+    chat_completion = client.chat.completions.create(
+        messages=[
+            {"role": "system", "content": system_prompt},
+            {"role": "user", "content": user_prompt},
+        ],
+        seed=42,
+        model="mistral",
+        n=3,
+        temperature=temperature,
+        top_p=top_p,
     )
+    return json.loads(chat_completion.choices[0].message.content)
 
+def review_paper(paper_abstract, temperature, top_p):
+    system_prompt = "###System prompt:\nYou are a helpful, professional reviewer. You are reviewing a scientific paper. You should provide at least 2 contribution bullets. You should provide at least 2 strengths and 2 weaknesses of the paper. You should provide at least 2 suggestions to improve the paper. You should provide at least 3 questions to the authors.\n###OutputFomrat:\nYou should return your answer in JSON format. For example, {'contribution': ['This paper proposes a new method for ...', 'This paper proposes a new method for ...'], 'strengths': ['The paper is well written.', 'The paper is well written.'], 'weaknesses': ['The paper is not well written.', 'The paper is not well written.'], 'suggestions': ['The authors should ...', 'The authors should ...'], 'questions': ['Why do you ...?', 'Why do you ...?']}"
+    user_prompt = "###Paper abstract:\n" + paper_abstract
+    result = generate(user_prompt, system_prompt, temperature, top_p)
+    return json.dumps(result, indent=4, ensure_ascii=False)
 
-    # List models API
-    models = client.models.list()
-    return str(models.model_dump())
 
+# 界面
+with gr.Blocks("Paper Abstract"):
+        paper_abstract = gr.inputs.Textbox(lines=10, label="Paper Abstract")
+with gr.Blocks("Temperature"):
+    temperature = gr.inputs.Slider(minimum=0, maximum=1, default=0.5, label="Temperature")
+with gr.Blocks("Top P"):
+    top_p = gr.inputs.Slider(minimum=0, maximum=1, default=0.9, label="Top P")
+
+# 结果展示
+
+with gr.Blocks("Output"):
+    gr.outputs.Textbox(label="Output")
 
-def greet(name):
-    return "Hello " + name + "!!" + test_api()
 
-iface = gr.Interface(fn=greet, inputs="text", outputs="text")
-iface.launch()
+iface = gr.Interface(fn=review_paper, inputs=[paper_abstract, temperature, top_p], outputs="text", title="Review Paper", description="This is a tool to help you review a paper.")
+iface.launch()