Update app.py

app.py

@@ -8,6 +8,9 @@ from huggingface_hub import upload_file
 from typing import Optional, Tuple, List, Union
 import logging
 import shutil
+import base64
+from PIL import Image
+import io
 
 # Configure logging
 logging.basicConfig(
@@ -46,23 +49,48 @@ class ImageToDxfConverter:
             'temp_debug': output_folder / "_debug.png"
         }
 
+    def _save_base64_image(self, base64_data: str) -> str:
+        """Save base64 image data to temporary file."""
+        try:
+            # Extract actual base64 data after comma
+            img_data = base64_data.split(',')[1]
+            img_bytes = base64.b64decode(img_data)
+            
+            # Create temporary file
+            with tempfile.NamedTemporaryFile(delete=False, suffix='.png') as temp_file:
+                temp_file.write(img_bytes)
+                return temp_file.name
+                
+        except Exception as e:
+            logger.error(f"Failed to save base64 image: {str(e)}")
+            return None
+
     def convert_image(self, 
-                     image_path: Optional[str], 
+                     image_input: Union[str, None], 
                      use_lines: bool = False) -> Tuple[Optional[str], Optional[str], List[str]]:
         """Convert image to DXF format."""
         try:
             # Input validation
-            if not image_path:
+            if not image_input:
                 return None, None, []
 
-            # Setup output directory with fixed path
+            # Handle base64 data
+            if isinstance(image_input, str) and image_input.startswith('data:image'):
+                temp_image_path = self._save_base64_image(image_input)
+                if not temp_image_path:
+                    return None, None, []
+                image_path = temp_image_path
+            else:
+                image_path = image_input
+
+            # Setup output directory
             output_dir = self._ensure_directory("OutputPDF")
             timestamp = datetime.datetime.now().strftime("%Y-%m-%d_%H-%M-%S")
             paths = self._generate_output_paths(output_dir, timestamp)
 
             # Prepare conversion command
             command = [
-                f"./{self.executable_path}",  # Use relative path with ./
+                f"./{self.executable_path}",
                 f"--imagePath={image_path}",
                 f"--outputPath={paths['temp_dxf']}",
                 f"--debug-output={paths['temp_debug']}"
@@ -124,6 +152,13 @@ class ImageToDxfConverter:
                 except Exception as e:
                     logger.error(f"Upload failed: {str(e)}")
 
+            # Clean up temporary file if it was created from base64
+            if isinstance(image_input, str) and image_input.startswith('data:image'):
+                try:
+                    os.unlink(image_path)
+                except Exception as e:
+                    logger.error(f"Failed to clean up temporary file: {str(e)}")
+
             return (
                 str(paths['output_dxf']),
                 str(paths['debug_png']) if use_lines and os.path.exists(paths['debug_png']) else None,
@@ -142,6 +177,7 @@ def create_gradio_interface():
         gr.Markdown("""
         # Image to DXF Converter
         Convert your images to DXF format for CAD software.
+        Press Ctrl+V to paste image from clipboard.
         """)
         
         with gr.Row():
@@ -149,7 +185,8 @@ def create_gradio_interface():
                 image_input = gr.Image(
                     type="filepath",
                     label="Input Image",
-                    elem_id="image_input"
+                    elem_id="image_input",
+                    height=300
                 )
             with gr.Column(scale=1):
                 use_lines_checkbox = gr.Checkbox(
@@ -159,7 +196,8 @@ def create_gradio_interface():
                 )
                 convert_btn = gr.Button(
                     "Convert to DXF",
-                    variant="primary"
+                    variant="primary",
+                    elem_id="convert_btn"
                 )
 
         with gr.Row():
@@ -172,10 +210,53 @@ def create_gradio_interface():
                 debug_output = gr.Image(
                     type="filepath",
                     label="Debug Preview",
-                    elem_id="debug_output"
+                    elem_id="debug_output",
+                    height=300
                 )
 
-        # Event handler
+        # Thêm JavaScript để xử lý paste từ clipboard
+        demo.load(js="""
+        function setupPasteHandler() {
+            document.addEventListener('paste', async function(e) {
+                e.preventDefault();
+                const items = e.clipboardData.items;
+                
+                for (let i = 0; i < items.length; i++) {
+                    if (items[i].type.indexOf('image') !== -1) {
+                        const blob = items[i].getAsFile();
+                        const reader = new FileReader();
+                        
+                        reader.onload = function(event) {
+                            const base64data = event.target.result;
+                            const imgElement = document.querySelector('#image_input img');
+                            if (imgElement) {
+                                imgElement.src = base64data;
+                                
+                                // Trigger conversion after image is loaded
+                                imgElement.onload = function() {
+                                    const convertBtn = document.querySelector('#convert_btn');
+                                    if (convertBtn) {
+                                        convertBtn.click();
+                                    }
+                                }
+                            }
+                        };
+                        
+                        reader.readAsDataURL(blob);
+                        break;
+                    }
+                }
+            });
+        }
+        
+        if (document.readyState === 'complete') {
+            setupPasteHandler();
+        } else {
+            window.addEventListener('load', setupPasteHandler);
+        }
+        """)
+
+        # Event handlers
         convert_btn.click(
             fn=converter.convert_image,
             inputs=[image_input, use_lines_checkbox],
@@ -188,9 +269,12 @@ def main():
     """Main entry point with proper error handling."""
     try:
         demo = create_gradio_interface()
+        demo.queue()  # Enable queuing for better handling of concurrent requests
         demo.launch(
             server_name="0.0.0.0",
-            server_port=7860
+            server_port=7860,
+            show_api=False,  # Disable API documentation page
+            share=False,     # Disable public URL generation
         )
     except Exception as e:
         logger.critical(f"Application failed to start: {str(e)}")