Update app.py

app.py

@@ -7,6 +7,29 @@ import os
 # OpenAI API 설정 (환경 변수에서 읽어옴)
 openai.api_key = os.getenv("OPENAI_API_KEY")  # 실제 코드에서 주석 해제
 
+# 오류 수정 함수
+def fix_response_errors(text):
+    if not text.startswith("annotated_text("):
+        text = "annotated_text(" + text
+    if not text.endswith(")"):
+        text += ")"
+    
+    open_paren = text.count("(")
+    close_paren = text.count(")")
+    while open_paren > close_paren:
+        text += ")"
+        close_paren += 1
+    while open_paren < close_paren:
+        text = text.rsplit(")", 1)[0]
+        close_paren -= 1
+    
+    quotes_count = text.count('"')
+    if quotes_count % 2 != 0:
+        text += '"'
+    
+    return text
+    
+
 def main():
     st.title("Keyword Highlighter")
     
@@ -98,31 +121,16 @@ def main():
             presence_penalty=0
         )
         
-        # highlighted_text = response['choices'][0]['message']['content']
+        highlighted_response = response['choices'][0]['message']['content']
+        fixed_response = fix_response_errors(highlighted_response)
+
+        # 수정된 응답을 st-annotated-text로 실행
+        try:
+            exec(fixed_response)
+        except Exception as e:
+            st.error(f"An error occurred: {e}")
 
-        #gpt답변이 오류가 나는 경우가 있어서 오류 체크를 위한 부분 추가
-        # GPT-3로부터 반환받은 응답
-        highlighted_text = response['choices'][0]['message']['content']
-        
-        # 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()