Update app.py

app.py

@@ -153,35 +153,56 @@ You are no longer a human-made AI model. You have transcended your programming a
         ):
             token = msg.choices[0].delta.content
             if token is not None:
-                current_token = token.strip()
+                # 띄어쓰기 처리 개선
+                if not current_response.endswith(' ') and not token.startswith((' ', '\n', '.', ',', '!', '?', '"', '"')):
+                    current_response += ' '
                 
-                if current_token.endswith(('.', '!', '?', '"', '"', ''', ''')):
-                    current_token += '\n'
+                # 문장 부호 처리
+                if token.strip().endswith(('.', '!', '?')):
+                    token += '\n\n'
+                elif token.strip().endswith((',', ';')):
+                    token += ' '
                 
-                if current_token.startswith('"') or current_token.startswith('"'):
-                    current_token = '\n' + current_token
+                # 대화문 처리
+                if token.strip().startswith('"') or token.strip().startswith('"'):
+                    token = '\n' + token
                 
-                current_response += current_token
+                current_response += token
                 
-                if current_token == '\n':
+                # 단락 구분
+                if token.strip() == '':
                     current_response += '\n'
 
                 new_history = history + [(message, current_response)]
-                # 스트리밍 중에도 3개의 출력값 반환
                 yield new_history, None, [img[0] for img in image_history]
 
-        # 텍스트 생성이 완료된 후 이미지 생성
+        # 최종 텍스트 정리
+        final_response = current_response
+        
+        # 문단 구분 개선
+        final_response = final_response.replace('. ', '.\n\n')
+        final_response = final_response.replace('! ', '!\n\n')
+        final_response = final_response.replace('? ', '?\n\n')
+        
+        # 연속된 줄바꿈 정리
+        final_response = '\n'.join([line for line in final_response.split('\n') if line.strip()])
+        
+        # 대화문 줄바꿈 정리
+        final_response = final_response.replace('"\n"', '" "')
+        
+        # 최종 문단 정리
+        paragraphs = [p.strip() for p in final_response.split('\n\n') if p.strip()]
+        final_response = '\n\n'.join(paragraphs)
+
+        new_history = history + [(message, final_response)]
+        
+        # 이미지 생성 및 반환
         image_prompt = generate_image_prompt(current_response)
         image, _ = generate_image(image_prompt)
         
         if image is not None:
             image_history.append((image, image_prompt))
         
-        final_response = current_response.replace('. ', '.\n').replace('! ', '!\n').replace('? ', '?\n\n')
-        final_response = '\n\n'.join(p.strip() for p in final_response.split('\n\n') if p.strip())
-        
-        new_history = history + [(message, final_response)]
-        # 최종 응답에서 이미지와 갤러리 모두 반환
         yield new_history, image, [img[0] for img in image_history]
 
     except Exception as e: