update

backend/app/main.py

@@ -179,23 +179,36 @@ async def generate_ai_content(request: AIContentRequest, db: Session = Depends(g
 # ---- Asset Management ----
 
 @app.get("/api/assets/{asset_id}/status")
-async def get_asset_status(asset_id: int, db: Session = Depends(get_db)):
+async def get_asset_status(asset_id, db: Session = Depends(get_db)):
     """Get the analysis status of an asset"""
     try:
+        # Convert asset_id to int (handle large numbers from CockroachDB)
+        try:
+            asset_id = int(asset_id)
+        except (ValueError, TypeError):
+            raise HTTPException(status_code=400, detail=f"Invalid asset ID: {asset_id}")
+        
         from app.models import Asset
         conn = get_direct_psycopg2_connection()
-        if conn:
+        if not conn:
+            raise HTTPException(status_code=500, detail="Database connection failed")
+        
+        try:
+            cursor = conn.cursor()
+            # First check if extracted_content column exists
             try:
-                cursor = conn.cursor()
-                # First check if extracted_content column exists
                 cursor.execute("""
                     SELECT column_name 
                     FROM information_schema.columns 
                     WHERE table_name='assets' AND column_name='extracted_content'
                 """)
                 has_extracted_content = cursor.fetchone() is not None
-                
-                # Build query based on column existence
+            except Exception as col_check_error:
+                print(f"Column check error (non-fatal): {col_check_error}")
+                has_extracted_content = False
+            
+            # Build query based on column existence
+            try:
                 if has_extracted_content:
                     cursor.execute("""
                         SELECT id, name, analysis_status, analyzed_at, extracted_content
@@ -208,38 +221,52 @@ async def get_asset_status(asset_id: int, db: Session = Depends(get_db)):
                         FROM assets 
                         WHERE id = %s
                     """, (asset_id,))
-                
-                row = cursor.fetchone()
+            except Exception as query_error:
+                print(f"Query error for asset_id {asset_id}: {query_error}")
                 cursor.close()
                 conn.close()
-                
-                if row:
-                    result = {
-                        "asset_id": row[0],
-                        "name": row[1],
-                        "status": row[2] or "pending",
-                        "analyzed_at": row[3].isoformat() if row[3] else None,
-                    }
-                    # Add extracted_content only if column exists and value is present
-                    if has_extracted_content and len(row) > 4:
-                        result["extracted_content"] = row[4]
-                    else:
-                        result["extracted_content"] = None
-                    return result
+                raise HTTPException(status_code=500, detail=f"Query failed: {str(query_error)}")
+            
+            row = cursor.fetchone()
+            cursor.close()
+            conn.close()
+            
+            if row:
+                result = {
+                    "asset_id": row[0],
+                    "name": row[1],
+                    "status": row[2] or "pending",
+                    "analyzed_at": row[3].isoformat() if row[3] else None,
+                }
+                # Add extracted_content only if column exists and value is present
+                if has_extracted_content and len(row) > 4:
+                    result["extracted_content"] = row[4]
                 else:
-                    raise HTTPException(status_code=404, detail="Asset not found")
-            except HTTPException:
-                raise
-            except Exception as e:
-                if conn:
+                    result["extracted_content"] = None
+                return result
+            else:
+                # Log for debugging
+                print(f"Asset not found: id={asset_id}, type={type(asset_id)}")
+                raise HTTPException(status_code=404, detail=f"Asset not found: {asset_id}")
+        except HTTPException:
+            raise
+        except Exception as e:
+            if conn:
+                try:
+                    cursor.close()
                     conn.close()
-                print(f"Error in get_asset_status: {e}")
-                raise HTTPException(status_code=500, detail=f"Database error: {str(e)}")
-        raise HTTPException(status_code=500, detail="Database connection failed")
+                except:
+                    pass
+            print(f"Error in get_asset_status for asset_id {asset_id}: {e}")
+            import traceback
+            print(traceback.format_exc())
+            raise HTTPException(status_code=500, detail=f"Database error: {str(e)}")
     except HTTPException:
         raise
     except Exception as e:
-        print(f"Error in get_asset_status (outer): {e}")
+        print(f"Error in get_asset_status (outer) for asset_id {asset_id}: {e}")
+        import traceback
+        print(traceback.format_exc())
         raise HTTPException(status_code=500, detail=str(e))
 
 async def analyze_asset_background(asset_id: int, file_path: str, file_type: str):
@@ -448,11 +475,11 @@ async def upload_asset(
                             ))
                             row = cursor.fetchone()
                             conn.commit()
-                            cursor.close()
-                            conn.close()
                             if row:
-                                db_asset.id = row[0]
+                                db_asset.id = int(row[0])  # Ensure ID is int
                                 db_asset.created_at = row[1]
+                            cursor.close()
+                            conn.close()
                         except Exception as psycopg2_error:
                             print(f"Direct psycopg2 insert failed: {psycopg2_error}")
                             if conn:

frontend/src/pages/Repository.jsx

@@ -279,12 +279,18 @@ export default function Repository() {
   const pollAssetStatus = async (assetId, fileName) => {
     const maxAttempts = 60; // Poll for up to 60 seconds
     let attempts = 0;
+    let consecutive404s = 0;
+    const max404s = 5; // Allow up to 5 consecutive 404s (asset might not be visible yet)
+    
+    // Small delay before first poll to allow database commit
+    await new Promise(resolve => setTimeout(resolve, 500));
     
     while (attempts < maxAttempts) {
       try {
         const response = await fetch(`/api/assets/${assetId}/status`);
         if (response.ok) {
           const status = await response.json();
+          consecutive404s = 0; // Reset 404 counter on success
           setUploadProgress(prev => ({
             ...prev,
             [fileName]: {
@@ -297,9 +303,28 @@ export default function Repository() {
           if (status.status === 'completed' || status.status === 'failed') {
             break;
           }
+        } else if (response.status === 404) {
+          consecutive404s++;
+          // If we get too many 404s, stop polling (asset might not exist)
+          if (consecutive404s >= max404s) {
+            console.warn(`Asset ${assetId} not found after ${max404s} attempts`);
+            setUploadProgress(prev => ({
+              ...prev,
+              [fileName]: {
+                status: 'pending',
+                message: 'Asset status unavailable'
+              }
+            }));
+            break;
+          }
+          // For first few 404s, keep trying (might be replication lag)
+        } else {
+          // Other errors - log but keep trying
+          console.error(`Status check failed: ${response.status}`);
         }
       } catch (error) {
         console.error('Error polling status:', error);
+        // On network errors, keep trying
       }
       
       await new Promise(resolve => setTimeout(resolve, 1000)); // Poll every second