error control

app.py

@@ -20,43 +20,47 @@ vectorstore = FAISS.load_local(save_path, embedding_model, allow_dangerous_deser
 
 def chatbot(input_question, eng_trans=True, num_ref=3):
     """챗봇 함수"""
-    retriever = vectorstore.as_retriever(search_kwargs={"k": num_ref})
-
-    # 한국어로 질의
-    if not eng_trans:
-        basic_docs = retriever.invoke(input_question)
-    # 영어로 번역 후 질의(영어, 한글 둘 다 검색)
-    else:
-        eng = translate_ko_to_en(input_question)
-        basic_docs = retriever.invoke(input_question)
-        eng_docs = retriever.invoke(eng)
-        basic_docs = basic_docs + eng_docs
-
-    context = "\n".join([doc.page_content for doc in basic_docs])
-
-    client = OpenAI(api_key=YOUR_OPENAI_API_KEY)
-
-    # GPT-4 or GPT-4o-mini 등 모델 지정
-    completion = client.chat.completions.create(
-        model="gpt-4o-mini",
-        messages=[
-            {
-                "role": "user",
-                "content": f"""당신은 반도체와 차세대반도체학과에 대해서 설명하는 Assistant입니다.
-    차세대반도체학과는 서울의 중앙대학교 창의ICT공과대학에 설립된 학과입니다. 학과장 및 단장은 송상헌 교수이며 현재 서포터즈 3기 팀장은 정규단 학생입니다.
-    이 학과는 서울대학교, 중앙대학교, 포항공과대학교, 숭실대학교, 강원대학교, 대구대학교, 조선이공대학교가 학점교류를 통해 수강할 수 있는 학과입니다.
-    다음 맥락에 맞게 질문에 한글로 답하세요. 답을 모르겠다면 semicon@cau.ac.kr 또는 02-881-7301 로 문의하라고 하세요.
-
-    맥락: {context}
-
-    질문: {input_question}
-    """
-            }
-        ]
-    )
-
-    return completion.choices[0].message.content
-
+    try:
+        retriever = vectorstore.as_retriever(search_kwargs={"k": num_ref})
+    
+        # 한국어로 질의
+        if not eng_trans:
+            basic_docs = retriever.invoke(input_question)
+        # 영어로 번역 후 질의(영어, 한글 둘 다 검색)
+        else:
+            eng = translate_ko_to_en(input_question)
+            basic_docs = retriever.invoke(input_question)
+            eng_docs = retriever.invoke(eng)
+            basic_docs = basic_docs + eng_docs
+    
+        context = "\n".join([doc.page_content for doc in basic_docs])
+    
+        client = OpenAI(api_key=YOUR_OPENAI_API_KEY)
+    
+        # GPT-4 or GPT-4o-mini 등 모델 지정
+        completion = client.chat.completions.create(
+            model="gpt-4o-mini",
+            messages=[
+                {
+                    "role": "user",
+                    "content": f"""당신은 반도체와 차세대반도체학과에 대해서 설명하는 Assistant입니다.
+        차세대반도체학과는 서울의 중앙대학교 창의ICT공과대학에 설립된 학과입니다. 학과장 및 단장은 송상헌 교수이며 현재 서포터즈 3기 팀장은 정규단 학생입니다.
+        이 학과는 서울대학교, 중앙대학교, 포항공과대학교, 숭실대학교, 강원대학교, 대구대학교, 조선이공대학교가 학점교류를 통해 수강할 수 있는 학과입니다.
+        다음 맥락에 맞게 질문에 한글로 답하세요. 답을 모르겠다면 semicon@cau.ac.kr 또는 02-881-7301 로 문의하라고 하세요.
+    
+        맥락: {context}
+    
+        질문: {input_question}
+        """
+                }
+            ]
+        )
+    
+        return completion.choices[0].message.content
+    except openai.error.RateLimitError:
+        return 'API 사용 비용을 초과하여 에러가 났습니다. 관리자에게 문의하십시오.'
+    except Exception as e:
+        return f'오류 발생: {str(e)}'
 # Gradio Blocks 레이아웃으로 꾸민 버전
 with gr.Blocks() as demo:
     # 상단 이미지