Update app.py

app.py

@@ -600,43 +600,37 @@ def generate_report(title, info, progress=gr.Progress()):
         
         progress(0.5, desc="AI 모델 처리 중...")
         
-        try:
-            response = hf_client.text_generation(
-                prompt,
-                max_new_tokens=4000,
-                temperature=0.3,
-                repetition_penalty=1.2
-            )
-            
-            progress(0.8, desc="결과 처리 중...")
-            
-            if response:
-                formatted_response = f"""### AI 리포팅 결과\n\n{response}"""
-                return [
-                    gr.update(value=formatted_response, visible=True),
-                    gr.update(value="접기", visible=True)
-                ]
-            else:
-                return [
-                    gr.update(value="리포팅 생성에 실패했습니다.", visible=True),
-                    gr.update(value="접기", visible=True)
-                ]
+        # 최대 3번 재시도
+        max_retries = 3
+        for attempt in range(max_retries):
+            try:
+                response = hf_client.text_generation(
+                    prompt,
+                    max_new_tokens=2000,  # 토큰 수 줄임
+                    temperature=0.7,
+                    repetition_penalty=1.2,
+                    return_full_text=False,
+                    max_time=180  # 타임아웃 180초로 증가
+                )
                 
-        except Exception as e:
-            print(f"Model error: {str(e)}")
-            return [
-                gr.update(value="AI 모델 처리 중 오류가 발생했습니다.", visible=True),
-                gr.update(value="접기", visible=True)
-            ]
+                if response:
+                    progress(1.0, desc="완료!")
+                    formatted_response = f"### AI 리포팅\n\n{response}"
+                    return [formatted_response, gr.update(visible=True)]
+                    
+            except Exception as e:
+                if attempt < max_retries - 1:
+                    print(f"Retry {attempt + 1}/{max_retries} after error: {str(e)}")
+                    time.sleep(5)  # 5초 대기 후 재시도
+                    continue
+                else:
+                    raise e
+        
+        return ["리포팅 생성에 실패했습니다. 다시 시도해주세요.", gr.update(visible=True)]
             
     except Exception as e:
-        print(f"Report generation error: {str(e)}")
-        return [
-            gr.update(value="리포팅 생성 중 오류가 발생했습니다.", visible=True),
-            gr.update(value="접기", visible=True)
-        ]
-    finally:
-        progress(1.0, desc="완료!")
+        print(f"Model error: {str(e)}")
+        return ["AI 모델 처리 중 오류가 발생했습니다. 잠시 후 다시 시도해주세요.", gr.update(visible=True)]
 
 def toggle_report(report_content, show_report):
     """리포트 표시/숨김 토글"""