Update app.py

app.py

@@ -17,17 +17,20 @@ def analyze_book_cover(image):
     if image is None:
         return "Please upload an image.", "", "", "", ""
 
+    tmp_path = None
     try:
         # Save PIL Image to temporary file (analyzer expects file path)
         with tempfile.NamedTemporaryFile(delete=False, suffix='.jpg') as tmp:
-            image.save(tmp.name)
+            image.save(tmp.name, format='JPEG')
             tmp_path = tmp.name
 
         # Analyze
         result = analyzer.analyze(tmp_path)
 
-        # Clean up temp file
-        os.unlink(tmp_path)
+        # Check if analysis succeeded
+        if not result.get('success', False):
+            error = result.get('error', 'Unknown error')
+            return f"Analysis failed: {error}", "", "", "", ""
 
         # Extract interpretation
         interp = result.get('interpretation', {})
@@ -39,22 +42,28 @@ def analyze_book_cover(image):
 
         # Format detected elements
         image_regions = result.get('image_regions', [])
-        elements_text = f"**Detected {len(image_regions)} visual elements:**\n"
-        for elem in image_regions[:5]:  # Top 5
-            elements_text += f"- {elem['class_name']}: {elem['confidence']:.1%}\n"
+        if image_regions:
+            elements_text = f"**Detected {len(image_regions)} visual elements:**\n"
+            for elem in image_regions[:5]:  # Top 5
+                elements_text += f"- {elem['location']}: {elem['pytorch_class']} ({elem['confidence']:.1%})\n"
+        else:
+            elements_text = "No visual elements detected (possibly plain text cover)"
 
         # Format detected text
         text_regions = result.get('text_regions', [])
-        text_output = f"**Detected {len(text_regions)} text regions:**\n"
-        for region in text_regions[:10]:  # Top 10
-            text_output += f"- {region['text']}\n"
+        if text_regions:
+            text_output = f"**Detected {len(text_regions)} text regions:**\n"
+            for region in text_regions[:10]:  # Top 10
+                text_output += f"- {region['text']}\n"
 
-        # Format other text (reviews/quotes)
-        other = interp.get('other_text', [])
-        if other:
-            text_output += f"\n**Reviews/Quotes:**\n"
-            for text in other[:3]:
-                text_output += f"- {text}\n"
+            # Format other text (reviews/quotes)
+            other = interp.get('other_text', [])
+            if other:
+                text_output += f"\n**Reviews/Quotes:**\n"
+                for text in other[:3]:
+                    text_output += f"- {text}\n"
+        else:
+            text_output = "No text detected on cover"
 
         return title, authors, publisher, elements_text, text_output
 
@@ -62,6 +71,14 @@ def analyze_book_cover(image):
         error_msg = f"Error analyzing image: {str(e)}"
         return error_msg, "", "", "", ""
 
+    finally:
+        # Always clean up temp file, even if there's an error
+        if tmp_path and os.path.exists(tmp_path):
+            try:
+                os.unlink(tmp_path)
+            except:
+                pass  # Ignore cleanup errors
+
 # Create Gradio interface
 demo = gr.Interface(
     fn=analyze_book_cover,