Update app.py

app.py

@@ -245,12 +245,36 @@ class Demo:
         return []
 
 def remove_code_block(text):
-    pattern = r'```html\n(.+?)\n```'
-    match = re.search(pattern, text, re.DOTALL)
-    if match:
-        return match.group(1).strip()
-    else:
-        return text.strip()
+    # Remove markdown code block syntax
+    text = re.sub(r'```[python|html]?\n', '', text)
+    text = re.sub(r'\n```', '', text)
+    
+    # Remove duplicate imports and launch configurations
+    lines = text.split('\n')
+    filtered_lines = []
+    seen_imports = set()
+    
+    for line in lines:
+        # Skip empty lines
+        if not line.strip():
+            continue
+            
+        # Skip duplicate imports
+        if line.startswith('import ') or line.startswith('from '):
+            import_key = line.split('#')[0].strip()
+            if import_key in seen_imports:
+                continue
+            seen_imports.add(import_key)
+            
+        # Skip duplicate launch configurations    
+        if 'if __name__ == "__main__":' in line:
+            continue
+        if 'demo.launch()' in line:
+            continue
+            
+        filtered_lines.append(line)
+        
+    return '\n'.join(filtered_lines)
 
 def history_render(history: History):
     return gr.update(open=True), history