Update app.py

app.py

@@ -145,7 +145,7 @@ def get_package_name(import_name):
         return None
     return IMPORT_TO_PACKAGE.get(base_import, base_import)
 
-def analyze_code(code: str) -> str:
+def analyze_code(code: str, query: str = "") -> str:
     """코드 분석 결과를 HTML 형식으로 반환"""
     analysis = []
     
@@ -239,6 +239,38 @@ def analyze_code(code: str) -> str:
             analysis.append(f"<li><strong>{component}</strong></li>")
         analysis.append("</ul>")
     
+    # 4. 프롬프트 대응 분석
+    if query:
+        analysis.append("<h2>🧠 프롬프트 대응 분석</h2>")
+        analysis.append(f"<p>요청하신 <strong>\"{query}\"</strong>에 대한 대응은 다음과 같습니다:</p>")
+        
+        # 주요 기능 분석
+        features = []
+        if "계산기" in query or "계산" in query:
+            if "BMI" in code:
+                features.append("BMI 계산 기능")
+            if "단위 변환" in query and ("convert" in code or "변환" in code):
+                features.append("단위 변환 기능")
+        if "달력" in query and "calendar" in code:
+            features.append("달력 표시 기능")
+        if "메모" in query and ("저장" in code or "save" in code):
+            features.append("메모 저장 및 관리 기능")
+        if "타이머" in query and ("timer" in code or "타이머" in code):
+            features.append("타이머 기능")
+            
+        # 일반적인 분석 추가
+        features.append("직관적인 사용자 인터페이스")
+        features.append("사용자 입력 검증 및 오류 처리")
+        features.append("모던한 디자인과 레이아웃")
+        
+        if features:
+            analysis.append("<ul>")
+            for feature in features:
+                analysis.append(f"<li>{feature}</li>")
+            analysis.append("</ul>")
+            
+        analysis.append("<p>이 코드는 Gradio를 활용하여 요청하신 기능을 구현했으며, 사용자 편의성과 직관적인 UI를 갖추고 있습니다.</p>")
+    
     return "\n".join(analysis)
 
 async def try_claude_api(system_message, claude_messages, timeout=15):
@@ -450,12 +482,15 @@ def deploy_to_huggingface(code: str, token: str):
         
 class Demo:
     def __init__(self):
-        pass
-
+        self.current_query = ""
+        
     async def generation_code(self, query: Optional[str], _setting: Dict[str, str], _history: Optional[History]):
         if not query or query.strip() == '':
             query = random.choice(DEMO_LIST)['description']
         
+        # 현재 쿼리 저장
+        self.current_query = query
+        
         if _history is None:
             _history = []
         
@@ -480,7 +515,8 @@ class Demo:
             collected_content = None
             try:
                 async for content in try_claude_api(system_message, claude_messages):
-                    code = content
+                    # 코드 블록 표시 제거
+                    code = remove_code_block(content)
                     yield code
                 collected_content = code
                 
@@ -488,7 +524,8 @@ class Demo:
                 print(f"Falling back to OpenAI API due to Claude error: {str(claude_error)}")
                 
                 async for content in try_openai_api(openai_messages):
-                    code = content
+                    # 코드 블록 표시 제거
+                    code = remove_code_block(content)
                     yield code
                 collected_content = code
 
@@ -500,13 +537,17 @@ class Demo:
             raise ValueError(f'Error calling APIs: {str(e)}')
 
     def clear_history(self):
+        self.current_query = ""
         return []
+    
+    def get_current_query(self):
+        return self.current_query
 
 # 예제 프롬프트
 example_prompts = [
-    "만년달력을 만들어주세요. 년도와 월을 입력하면 해당 월의 달력이 표시되는 앱입니다.",
+    "한글 입력시 음성 생성 TTS를 구글 gtts기반으로 생성하라.",
     "BMI 계산기를 만들어주세요. 키와 몸무게를 입력하면 BMI 지수와 비만도를 계산해주는 앱입니다.",
-    "간단한 메모장을 만들어주세요. 메모를 작성하고 저장하면 목록에 표시되며, 저장된 메모를 선택하여 수정하거나 삭제할 수 있습니다.",
+    "MBTI 진단 서비스: 10가지 질문과 답변 선택하면 16가지 유형 진단과 상세 설명을 하라",
     "단위 변환기를 만들어주세요. 길이(m, cm, km 등), 무게(kg, g 등), 온도(섭씨, 화씨) 등을 변환할 수 있는 앱입니다.",
     "포모도로 타이머를 만들어주세요. 25분 집중, 5분 휴식을 반복하는 타이머로, 사이클 횟수도 표시됩니다."
 ]
@@ -575,6 +616,7 @@ with gr.Blocks(css=css) as demo:
         "system": SystemPrompt,
     })
     is_generating = gr.State(False)
+    current_query = gr.State("")
 
     gr.HTML("""
     <div class="header">
@@ -617,7 +659,7 @@ with gr.Blocks(css=css) as demo:
                 with gr.TabItem("분석"):
                     code_analysis = gr.HTML(label="코드 분석")
             
-            # Box 대신 div로 변경
+            # Group으로 변경
             with gr.Group(elem_classes="deploy-section"):
                 gr.HTML("<h3>배포 설정</h3>")
                 hf_token = gr.Textbox(
@@ -636,14 +678,14 @@ with gr.Blocks(css=css) as demo:
     """)
 
     # 이벤트 핸들러
-    def start_generating():
-        return "🔄 코드 생성 중...", gr.update(interactive=False), gr.update(interactive=False), gr.update(interactive=False), True
-    
+    def start_generating(query):
+        return "🔄 코드 생성 중...", gr.update(interactive=False), gr.update(interactive=False), gr.update(interactive=False), True, query
+
     def end_generating():
         return "", gr.update(interactive=True), gr.update(interactive=True), gr.update(interactive=True), False
     
-    def update_code_analysis(code):
-        analysis = analyze_code(code)
+    def update_code_analysis(code, query):
+        analysis = analyze_code(code, query)
         return analysis
     
     def handle_boost(prompt):
@@ -652,14 +694,15 @@ with gr.Blocks(css=css) as demo:
     
     generate_btn.click(
         fn=start_generating,
-        outputs=[status_html, generate_btn, boost_btn, clear_btn, is_generating]
+        inputs=[input_text],
+        outputs=[status_html, generate_btn, boost_btn, clear_btn, is_generating, current_query]
     ).then(
         fn=demo_instance.generation_code,
         inputs=[input_text, setting, history],
         outputs=code_output
     ).then(
         fn=update_code_analysis,
-        inputs=[code_output],
+        inputs=[code_output, current_query],
         outputs=[code_analysis]
     ).then(
         fn=end_generating,