Update app.py

app.py

@@ -61,8 +61,16 @@ class URLProcessor:
             return {'is_valid': False, 'message': f'URL validation failed: {str(e)}'}
 
     def fetch_content(self, url: str) -> Optional[Dict]:
-        """Simple content fetcher that returns the raw HTML"""
+        """Universal content fetcher with special case handling"""
         try:
+            # Google Drive document handling
+            if 'drive.google.com' in url:
+                return self._handle_google_drive(url)
+    
+            # Google Calendar ICS handling
+            if 'calendar.google.com' in url and ' ics' in url:
+                return self._handle_google_calendar(url)
+    
             logger.info(f"Fetching content from URL: {url}")
             response = self.session.get(url, timeout=self.timeout)
             response.raise_for_status()
@@ -101,30 +109,61 @@ class URLProcessor:
                 url_list = [url.strip() for url in url_list if url.strip()]
     
                 for url in url_list:
-                    logger.info(f"Processing URL: {url}")
                     validation = processor.validate_url(url)
-                    
                     if validation.get('is_valid'):
-                        logger.info(f"URL {url} is valid, fetching content...")
                         content = processor.fetch_content(url)
-                        
                         if content:
-                            logger.info(f"Content fetched successfully from {url}") 
-                            results.append(content['content'])
-                        else:
-                            logger.warning(f"Failed to fetch content from {url}")
+                            # Convert HTML content to a proper JSON object
+                            url_data = {
+                                'source': 'url',
+                                'url': url,
+                                'content': content.get('content', ''),
+                                'content_type': content.get('content_type', ''),
+                                'timestamp': datetime.now().isoformat()
+                            }
+                            results.append(url_data)
                     else:
-                        logger.error(f"Invalid URL: {validation.get('message')}")
+                        logger.warning(f"Invalid URL: {url} - {validation.get('message')}")
     
-            # Combine results if needed
-            if combine:
-                combined_content = "\n".join(results)
-                return combined_content, results, None
+            # Process files if provided
+            if file:
+                file_results = file_processor.process_file(file)
+                if file_results:
+                    results.extend(file_results)
+    
+            # Generate QR codes
+            if results:
+                if combine:
+                    combined_data = {
+                        'type': 'combined_data',
+                        'items': results,
+                        'timestamp': datetime.now().isoformat()
+                    }
+                    qr_paths = generate_qr_code(combined_data, combined=True)
+                else:
+                    qr_paths = []
+                    for item in results:
+                        item_paths = generate_qr_code(item, combined=True)
+                        if item_paths:
+                            qr_paths.extend(item_paths)
+    
+                if qr_paths:
+                    return (
+                        results,
+                        [str(path) for path in qr_paths],
+                        f"✅ Successfully processed {len(results)} items and generated {len(qr_paths)} QR code(s)!"
+                    )
+                else:
+                    return None, [], "❌ Failed to generate QR codes. Please check the input data."
+            else:
+                return None, [], "⚠️ No valid content to process. Please provide some input data."
     
-            return results, [], None
         except Exception as e:
-            logger.error(f"Error processing inputs: {e}")
-            return None, [], f"❌ An error occurred: {str(e)}"
+            logger.error(f"Processing error: {e}")
+            import traceback
+            logger.error(traceback.format_exc())  # Print the full stack trace
+            return None, [], f"❌ Error: {str(e)}"
+
             
     def _handle_google_drive(self, url: str) -> Optional[Dict]:
         """Process Google Drive file links"""