Update app.py (#1)

- Update app.py (4fa8658e67b4b749f0ee3fabfe213fa1b279c757)


Co-authored-by: Yilun <yilunzhao@users.noreply.huggingface.co>

app.py

@@ -1,166 +1,3 @@
-# import gradio as gr
-# import json, os
-# import urllib.request
-
-# # Replace 'YOUR_API_KEY' with your actual YouTube Data API key
-# API_KEY = os.getenv('api_key')
-
-# def get_youtube_id(youtube_url):
-#     if 'youtube.com' in youtube_url:
-#         video_id = youtube_url.split('v=')[-1]
-#     elif 'youtu.be' in youtube_url:
-#         video_id = youtube_url.split('/')[-1].split('?')[0]
-#     return video_id
-
-# def check_cc_license(youtube_url):
-#     # Extract video ID from the URL
-#     video_id = get_youtube_id(youtube_url)
-    
-#     # YouTube Data API URL to get video details
-#     api_url = f'https://www.googleapis.com/youtube/v3/videos?id={video_id}&part=status&key={API_KEY}'
-    
-#     try:
-#         # Fetch video details
-#         response = urllib.request.urlopen(api_url)
-#         data = json.load(response)
-        
-#         # Check the license status
-#         for item in data['items']:
-#             if item['status']['license'] == 'creativeCommon':
-#                 return f"Yes."
-#             else:
-#                 return f"No."
-                
-#     except Exception as e:
-#         return f"An error occurred: {str(e)}"
-
-# # Gradio interface
-# interface = gr.Interface(
-#     fn=check_cc_license,
-#     inputs=gr.Textbox(label="YouTube Video URL"),
-#     outputs=gr.Textbox(label="Creative Commons license?")
-# )
-
-# if __name__ == "__main__":
-#     interface.launch()
-
-
-# import gradio as gr
-# import asyncio
-# import os
-# from openai import AsyncOpenAI
-
-# # 从您的模块中导入必要的函数
-# from utils.generate_distractors import prepare_q_inputs, construct_prompt_textonly, generate_distractors
-# from utils.api_utils import generate_from_openai_chat_completion
-# # 修改generate_distractors函数，使其成为异步函数
-# # 假设generate_distractors函数定义在您的模块中，我们需要修改它
-# # 如果无法修改原始模块，请在此处重新定义
-
-# async def generate_distractors_async(model_name: str, 
-#                                      queries: list,
-#                                      n: int=1,
-#                                      max_tokens: int=4096):
-#     assert model_name in ["gpt-4o-mini", "gpt-4-turbo", "gpt-4o", "gpt-4o-2024-08-06"], "Invalid model name"
-
-#     client = AsyncOpenAI(api_key=os.environ.get("OPENAI_API_KEY"), base_url="https://yanlp.zeabur.app/v1")
-#     messages = prepare_q_inputs(queries)
-
-#     # 直接等待协程而不是使用asyncio.run()
-#     responses = await generate_from_openai_chat_completion(
-#         client,
-#         messages=messages, 
-#         engine_name=model_name,
-#         n=n,
-#         max_tokens=max_tokens,
-#         requests_per_minute=30,
-#         json_format=True
-#     )
-
-#     for query, response in zip(queries, responses):
-#         new_options = response
-#         if new_options and "distractors" in new_options:
-#             query["option_5"] = new_options["distractors"].get("E", "")
-#             query["option_6"] = new_options["distractors"].get("F", "")
-#             query["option_7"] = new_options["distractors"].get("G", "")
-#             query["distractor_analysis"] = new_options["distractors"].get("analysis_of_distractors", "")
-#         else:
-#             query["option_5"] = ""
-#             query["option_6"] = ""
-#             query["option_7"] = ""
-#             query["distractor_analysis"] = ""
-
-#     return queries
-
-# # 定义异步处理函数
-# async def generate_distractors_gradio(question, option1, option2, option3, option4, answer, answer_analysis):
-#     query = {
-#         'question': question,
-#         'option_1': option1,
-#         'option_2': option2,
-#         'option_3': option3,
-#         'option_4': option4,
-#         'answer': answer,
-#         'answer_analysis': answer_analysis
-#     }
-
-#     queries = [query]  # 因为函数期望的是一个列表
-
-#     # 调用异步生成干扰项的函数
-#     results = await generate_distractors_async(
-#         model_name="gpt-4o-mini",
-#         queries=queries,
-#         n=1,
-#         max_tokens=4096
-#     )
-
-#     # 提取结果
-#     result = results[0]
-#     new_options = {
-#         'E': result.get('option_5', ''),
-#         'F': result.get('option_6', ''),
-#         'G': result.get('option_7', '')
-#     }
-#     distractor_analysis = result.get('distractor_analysis', '')
-
-#     # 返回新的干扰项和分析
-#     return new_options, distractor_analysis
-
-# # 创建Gradio界面
-# with gr.Blocks() as demo:
-#     gr.Markdown("# 多项选择题干扰项生成器")
-#     with gr.Row():
-#         question_input = gr.Textbox(label="问题", lines=2)
-#     with gr.Row():
-#         option1_input = gr.Textbox(label="选项A")
-#         option2_input = gr.Textbox(label="选项B")
-#     with gr.Row():
-#         option3_input = gr.Textbox(label="选项C")
-#         option4_input = gr.Textbox(label="选项D")
-#     with gr.Row():
-#         answer_input = gr.Textbox(label="正确答案")
-#     with gr.Row():
-#         answer_analysis_input = gr.Textbox(label="答案解析", lines=3)
-#     with gr.Row():
-#         generate_button = gr.Button("生成干扰项")
-#     with gr.Row():
-#         output_options = gr.JSON(label="生成的干扰选项")
-#     with gr.Row():
-#         output_analysis = gr.Textbox(label="干扰项解析", lines=5)
-
-#     # 定义按钮点击事件，注意这里不需要修改，Gradio会自动处理异步函数
-#     generate_button.click(
-#         fn=generate_distractors_gradio,
-#         inputs=[question_input, option1_input, option2_input, option3_input, option4_input, answer_input, answer_analysis_input],
-#         outputs=[output_options, output_analysis]
-#     )
-
-# # 运行Gradio应用
-# demo.launch()
-
-
-
-
 import gradio as gr
 import asyncio
 import os
@@ -172,7 +9,6 @@ from openai import AsyncOpenAI
 
 # 请确保在环境变量中设置了您的YouTube Data API密钥
 API_KEY = "AIzaSyDyPpkFRUpUuSMQbhxwTFxCBLK5qTHU-ms"
-os.environ["OPENAI_API_KEY"] = "sk-h8rybT55JORrT40y2397A06c8f2444F8Ad49407fE85eBf76"
 
 def get_youtube_id(youtube_url):
     if 'youtube.com' in youtube_url:
@@ -218,9 +54,6 @@ def check_cc_license(youtube_url):
 # 从您的模块中导入必要的函数
 from utils.generate_distractors import prepare_q_inputs, construct_prompt_textonly, generate_distractors
 from utils.api_utils import generate_from_openai_chat_completion
-# 修改generate_distractors函数，使其成为异步函数
-# 假设generate_distractors函数定义在您的模块中，我们需要修改它
-# 如果无法修改原始模块，请在此处重新定义
 
 async def generate_distractors_async(model_name: str, 
                                      queries: list,
@@ -259,6 +92,10 @@ async def generate_distractors_async(model_name: str,
 
 # 定义异步处理函数
 async def generate_distractors_gradio(question, option1, option2, option3, option4, answer, answer_analysis):
+    is_valid, message = validate_inputs(question, option1, option2, option3, option4, answer, answer_analysis)
+    if not is_valid:
+        return {"error": message}, ""  # Output error message
+    
     query = {
         'question': question,
         'option_1': option1,
@@ -273,7 +110,7 @@ async def generate_distractors_gradio(question, option1, option2, option3, optio
 
     # 调用异步生成干扰项的函数
     results = await generate_distractors_async(
-        model_name="gpt-4o-mini",
+        model_name="gpt-4o",
         queries=queries,
         n=1,
         max_tokens=4096
@@ -286,11 +123,28 @@ async def generate_distractors_gradio(question, option1, option2, option3, optio
         'F': result.get('option_6', ''),
         'G': result.get('option_7', '')
     }
+    new_option_str = f"E: {new_options['E']}\nF:{new_options['F']}\nG:{new_options['G']}"
     distractor_analysis = result.get('distractor_analysis', '')
 
     # 返回新的干扰项和分析
-    return new_options, distractor_analysis
-
+    return new_option_str, distractor_analysis
+
+def validate_inputs(question, option1, option2, option3, option4,  answer, analysis):
+    if not question:
+        return False, "问题不能为空"
+    if not option1:
+        return False, "选项A不能为空"
+    if not option2:
+        return False, "选项B不能为空"
+    if not option3:
+        return False, "选项C不能为空"
+    if not option4:
+        return False, "选项D不能为空"
+    if not answer:
+        return False, "正确答案不能为空"
+    if not analysis:
+        return False, "答案解析不能为空"
+    return True, ""
 
 with gr.Blocks() as demo:
     gr.Markdown("# CC检查器和干扰项生成器")
@@ -313,15 +167,14 @@ with gr.Blocks() as demo:
             with gr.Row():
                 option1_input = gr.Textbox(label="选项A")
                 option2_input = gr.Textbox(label="选项B")
-            with gr.Row():
                 option3_input = gr.Textbox(label="选项C")
-                option4_input = gr.Textbox(label="选项D(可以输入也可以不输入)")
+                option4_input = gr.Textbox(label="选项D")
             with gr.Row():
                 answer_input = gr.Textbox(label="正确答案（A/B/C/D）")
             with gr.Row():
                 answer_analysis_input = gr.Textbox(label="答案解析", lines=3)
             generate_button = gr.Button("生成干扰项")
-            output_options = gr.JSON(label="生成的干扰选项")
+            output_options = gr.Textbox(label="生成的干扰选项")
             output_analysis = gr.Textbox(label="干扰项解析", lines=5)
             generate_button.click(
                 fn=generate_distractors_gradio,
@@ -331,4 +184,3 @@ with gr.Blocks() as demo:
 
 # 运行Gradio应用
 demo.launch()
-