Update app.py

app.py

@@ -7,29 +7,6 @@ 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")
     
@@ -107,7 +84,7 @@ def main():
         # )"""
 
         # user_prompt = f"Based on the task description, annotate the following text by highlighting key words about the topic: {user_text}"
-        user_prompt = f"First, extract key words for the topic. Consider the context and structure of the paragraph. Then highlight the key words and output them in Python's st-annotated-text format: {user_text}"
+        user_prompt = f"First, extract key words for the topic st-annotated-text format.: {user_text}"
         
         messages = [{"role": "system", "content": task_description}, {"role": "user", "content": user_prompt}]
         
@@ -121,14 +98,11 @@ def main():
             presence_penalty=0
         )
         
-        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}")
+        highlighted_text = response['choices'][0]['message']['content']
+        
+        # 여기서는 간단하게 exec 함수를 이용해 GPT-3.5-turbo가 생성한 코드를 실행합니다.
+        # 실제 프로덕션 환경에서는 보안 이슈를 고려해야 합니다.
+        exec(highlighted_text)