Update app.py

app.py

@@ -98,11 +98,31 @@ def main():
             presence_penalty=0
         )
         
+        # highlighted_text = response['choices'][0]['message']['content']
+
+        #gpt답변이 오류가 나는 경우가 있어서 오류 체크를 위한 부분 추가
+        # GPT-3로부터 반환받은 응답
         highlighted_text = response['choices'][0]['message']['content']
         
-        # 여기서는 간단하게 exec 함수를 이용해 GPT-3.5-turbo가 생성한 코드를 실행합니다.
-        # 실제 프로덕션 환경에서는 보안 이슈를 고려해야 합니다.
-        exec(highlighted_text)
+        # 1. 반환된 텍스트가 `annotated_text(`로 시작하고 `)`로 끝나는지 확인
+        if not highlighted_text.startswith("annotated_text(") or not highlighted_text.endswith(")"):
+            st.error("Invalid response format from GPT-3.")
+        
+        # 2. 괄호의 개수가 올바른지 확인
+        elif highlighted_text.count("(") != highlighted_text.count(")"):
+            st.error("Mismatched parentheses in the response.")
+        
+        # 3. 문자열 내의 쌍따옴표나 싱글 따옴표에 문제가 없는지 확인
+        elif highlighted_text.count('"') % 2 != 0:
+            st.error("Mismatched quotes in the response.")
+        
+        else:
+            # 반환된 텍스트를 실행합니다. (이 부분은 보안상 위험할 수 있으므로 신중하게 사용해야 합니다.)
+            exec(highlighted_text)
+
+        # # 여기서는 간단하게 exec 함수를 이용해 GPT-3.5-turbo가 생성한 코드를 실행합니다.
+        # # 실제 프로덕션 환경에서는 보안 이슈를 고려해야 합니다.
+        # exec(highlighted_text)
 
 if __name__ == "__main__":
     main()