Update app.py

app.py

@@ -1532,65 +1532,6 @@ Provide specific solutions for each issue."""
                             error_msg += f" - {error_data['error']}"
                         elif 'message' in error_data:
                             error_msg += f" - {error_data['message']}"
-                except Exception as e:
-                    logger.error(f"Error parsing error response: {e}")
-                    error_msg += f" - {response.text[:200]}"
-                
-                yield f"❌ {error_msg}"
-                return
-            
-            buffer = ""
-            line_count = 0
-            
-            for line in response.iter_lines():
-                if not line:
-                    continue
-                
-                line_count += 1
-                
-                try:
-                    line_str = line.decode('utf-8').strip()
-                    
-                    # Skip non-SSE lines
-                    if not line_str.startswith("data: "):
-                        logger.debug(f"Skipping non-SSE line: {line_str[:50]}")
-                        continue
-                    
-                    data_str = line_str[6:]  # Remove "data: " prefix
-                    
-                    if data_str == "[DONE]":
-                        logger.info(f"Stream completed. Total lines: {line_count}")
-                        break
-                    
-                    if not data_str:
-                        continue
-                    
-                    # Parse JSON data
-                    try:
-                        data = json.loads(data_str)
-                    except json.JSONDecodeError as e:
-                        logger.warning(f"JSON decode error on line {line_count}: {e}")
-                        logger.debug(f"Problematic data: {data_str[:100]}")
-                        continue
-                    
-                    # Extract content from response
-                    if isinstance(data, dict) and "choices" in data:
-                        choices = data["choices"]
-                        if isinstance(choices, list) and len(choices) > 0:
-                            choice = choices[0]
-                            if isinstance(choice, dict) and "delta" in choice:
-                                delta = choice["delta"]
-                                if isinstance(delta, dict) and "content" in delta:
-                                    content = delta["content"]
-                                    if content:
-                                        buffer += content
-                                        
-                                        # Yield when buffer is large enough
-                                        if len(buffer) >= 50 or '\n' in buffer:
-                                            yield buffer
-                                            buffer = ""
-                                            time.sleep(0.01)
-                    
                 except Exception as e:
                     logger.error(f"Error processing line {line_count}: {str(e)}")
                     logger.debug(f"Problematic line: {line_str[:100] if 'line_str' in locals() else 'N/A'}")
@@ -1889,10 +1830,10 @@ You provide feedback that's critical yet encouraging."""
             if name_match:
                 extracted_name = name_match.group(1).strip()
                 # Remove markdown and extra characters
-                extracted_name = re.sub(r'[*:\s]+, '', extracted_name)
+                extracted_name = re.sub(r'[*:\s]+', '', extracted_name)
                 extracted_name = re.sub(r'^[*:\s]+', '', extracted_name)
                 # Remove age if it's part of the name
-                extracted_name = re.sub(r'\s*,?\s*\d+\s*(?:세|살)?, '', extracted_name)
+                extracted_name = re.sub(r'\s*,?\s*\d+\s*(?:세|살)?', '', extracted_name)
                 if extracted_name and len(extracted_name) > 1:
                     name = extracted_name
                     break
@@ -2007,7 +1948,66 @@ You provide feedback that's critical yet encouraging."""
                     value = re.sub(r'\*\*', '', value)
                     value = re.sub(r'^\s*[-•*]\s*', '', value)
                     # Remove trailing punctuation
-                    value = re.sub(r'[,.:;], '', value)
+                    value = re.sub(r'[,.:;]error(f"Error parsing error response: {e}")
+                    error_msg += f" - {response.text[:200]}"
+                
+                yield f"❌ {error_msg}"
+                return
+            
+            buffer = ""
+            line_count = 0
+            
+            for line in response.iter_lines():
+                if not line:
+                    continue
+                
+                line_count += 1
+                
+                try:
+                    line_str = line.decode('utf-8').strip()
+                    
+                    # Skip non-SSE lines
+                    if not line_str.startswith("data: "):
+                        logger.debug(f"Skipping non-SSE line: {line_str[:50]}")
+                        continue
+                    
+                    data_str = line_str[6:]  # Remove "data: " prefix
+                    
+                    if data_str == "[DONE]":
+                        logger.info(f"Stream completed. Total lines: {line_count}")
+                        break
+                    
+                    if not data_str:
+                        continue
+                    
+                    # Parse JSON data
+                    try:
+                        data = json.loads(data_str)
+                    except json.JSONDecodeError as e:
+                        logger.warning(f"JSON decode error on line {line_count}: {e}")
+                        logger.debug(f"Problematic data: {data_str[:100]}")
+                        continue
+                    
+                    # Extract content from response
+                    if isinstance(data, dict) and "choices" in data:
+                        choices = data["choices"]
+                        if isinstance(choices, list) and len(choices) > 0:
+                            choice = choices[0]
+                            if isinstance(choice, dict) and "delta" in choice:
+                                delta = choice["delta"]
+                                if isinstance(delta, dict) and "content" in delta:
+                                    content = delta["content"]
+                                    if content:
+                                        buffer += content
+                                        
+                                        # Yield when buffer is large enough
+                                        if len(buffer) >= 50 or '\n' in buffer:
+                                            yield buffer
+                                            buffer = ""
+                                            time.sleep(0.01)
+                    
+                except Exception as e:
+                    logger., '', value)
                     cleaned = value.strip()
                     if cleaned:
                         return cleaned
@@ -2395,7 +2395,66 @@ def format_screenplay_display(screenplay_text: str) -> str:
     
     # Format scene headings
     formatted_text = re.sub(
-        r'^(INT\.|EXT\.)(.*?),
+        r'^(INT\.|EXT\.)(.*?)error(f"Error parsing error response: {e}")
+                    error_msg += f" - {response.text[:200]}"
+                
+                yield f"❌ {error_msg}"
+                return
+            
+            buffer = ""
+            line_count = 0
+            
+            for line in response.iter_lines():
+                if not line:
+                    continue
+                
+                line_count += 1
+                
+                try:
+                    line_str = line.decode('utf-8').strip()
+                    
+                    # Skip non-SSE lines
+                    if not line_str.startswith("data: "):
+                        logger.debug(f"Skipping non-SSE line: {line_str[:50]}")
+                        continue
+                    
+                    data_str = line_str[6:]  # Remove "data: " prefix
+                    
+                    if data_str == "[DONE]":
+                        logger.info(f"Stream completed. Total lines: {line_count}")
+                        break
+                    
+                    if not data_str:
+                        continue
+                    
+                    # Parse JSON data
+                    try:
+                        data = json.loads(data_str)
+                    except json.JSONDecodeError as e:
+                        logger.warning(f"JSON decode error on line {line_count}: {e}")
+                        logger.debug(f"Problematic data: {data_str[:100]}")
+                        continue
+                    
+                    # Extract content from response
+                    if isinstance(data, dict) and "choices" in data:
+                        choices = data["choices"]
+                        if isinstance(choices, list) and len(choices) > 0:
+                            choice = choices[0]
+                            if isinstance(choice, dict) and "delta" in choice:
+                                delta = choice["delta"]
+                                if isinstance(delta, dict) and "content" in delta:
+                                    content = delta["content"]
+                                    if content:
+                                        buffer += content
+                                        
+                                        # Yield when buffer is large enough
+                                        if len(buffer) >= 50 or '\n' in buffer:
+                                            yield buffer
+                                            buffer = ""
+                                            time.sleep(0.01)
+                    
+                except Exception as e:
+                    logger.,
         r'**\1\2**',
         screenplay_text,
         flags=re.MULTILINE
@@ -2403,7 +2462,66 @@ def format_screenplay_display(screenplay_text: str) -> str:
     
     # Format character names (all caps on their own line)
     formatted_text = re.sub(
-        r'^([A-Z][A-Z\s]+),
+        r'^([A-Z][A-Z\s]+)error(f"Error parsing error response: {e}")
+                    error_msg += f" - {response.text[:200]}"
+                
+                yield f"❌ {error_msg}"
+                return
+            
+            buffer = ""
+            line_count = 0
+            
+            for line in response.iter_lines():
+                if not line:
+                    continue
+                
+                line_count += 1
+                
+                try:
+                    line_str = line.decode('utf-8').strip()
+                    
+                    # Skip non-SSE lines
+                    if not line_str.startswith("data: "):
+                        logger.debug(f"Skipping non-SSE line: {line_str[:50]}")
+                        continue
+                    
+                    data_str = line_str[6:]  # Remove "data: " prefix
+                    
+                    if data_str == "[DONE]":
+                        logger.info(f"Stream completed. Total lines: {line_count}")
+                        break
+                    
+                    if not data_str:
+                        continue
+                    
+                    # Parse JSON data
+                    try:
+                        data = json.loads(data_str)
+                    except json.JSONDecodeError as e:
+                        logger.warning(f"JSON decode error on line {line_count}: {e}")
+                        logger.debug(f"Problematic data: {data_str[:100]}")
+                        continue
+                    
+                    # Extract content from response
+                    if isinstance(data, dict) and "choices" in data:
+                        choices = data["choices"]
+                        if isinstance(choices, list) and len(choices) > 0:
+                            choice = choices[0]
+                            if isinstance(choice, dict) and "delta" in choice:
+                                delta = choice["delta"]
+                                if isinstance(delta, dict) and "content" in delta:
+                                    content = delta["content"]
+                                    if content:
+                                        buffer += content
+                                        
+                                        # Yield when buffer is large enough
+                                        if len(buffer) >= 50 or '\n' in buffer:
+                                            yield buffer
+                                            buffer = ""
+                                            time.sleep(0.01)
+                    
+                except Exception as e:
+                    logger.,
         r'**\1**',
         formatted_text,
         flags=re.MULTILINE
@@ -2871,5 +2989,4 @@ if __name__ == "__main__":
         server_port=7860,
         share=False,
         debug=True
-    )
-    
+    )