ui fetch and gen

app.py

@@ -59,22 +59,38 @@ async def get_rpk_info(filename: str):
         attrs_info = pyprt.get_rpk_attributes_info(rpk_path)
         
         formatted_attrs = []
-        for attr in attrs_info:
-            if hasattr(attr, 'get_name'):
-                formatted_attrs.append({
-                    "name": attr.get_name(),
-                    "type": str(attr.get_type()), 
-                    "defaultValue": attr.get_default_value(),
-                })
-            else:
-                # Fallback implementation if specific attribute object not returned
-                # This often happens if the RPK doesn't have exposed attributes or version mismatch
+        
+        # Handle Dictionary return type (common in some PyPRT versions)
+        if isinstance(attrs_info, dict):
+            for name, value in attrs_info.items():
+                # Determine type from value
+                attr_type = "string"
+                if isinstance(value, bool): attr_type = "bool"
+                elif isinstance(value, float): attr_type = "float"
+                elif isinstance(value, int): attr_type = "float" # Convert int to float for consistency
+                
                 formatted_attrs.append({
-                    "name": str(attr),
-                    "type": "string",
-                    "defaultValue": ""
+                    "name": name,
+                    "type": attr_type,
+                    "defaultValue": value
                 })
-            
+        # Handle List of Objects return type (standard PyPRT)
+        elif hasattr(attrs_info, '__iter__'):
+             for attr in attrs_info:
+                if hasattr(attr, 'get_name'):
+                    formatted_attrs.append({
+                        "name": attr.get_name(),
+                        "type": str(attr.get_type()), 
+                        "defaultValue": attr.get_default_value(),
+                    })
+                elif isinstance(attr, str):
+                    # Fallback if list of strings
+                    formatted_attrs.append({
+                        "name": attr,
+                        "type": "string",
+                        "defaultValue": ""
+                    })
+
         return {"attributes": formatted_attrs}
     except Exception as e:
         logger.error(f"Error inspecting RPK: {e}")